Skip to content

Commit aa8097f

Browse files
alxmrsclaude
andcommitted
Address review: check a given config's partition count
Co-Authored-By: Claude Opus 5.5 <[email protected]> Claude-Session: https://claude.ai/code/session_01NyN7idACeDMh94qrnFMu6h
1 parent 7ddadec commit aa8097f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

‎tests/test_sql.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@ def test_a_small_query_is_not_split_across_partitions():
3737

3838
def test_a_config_given_is_used_as_it_is():
3939
config = SessionConfig().with_target_partitions(4)
40-
assert "RepartitionExec" in _small_join_plan(xarray_sql.Context(config))
40+
plan = _small_join_plan(xarray_sql.Context(config))
41+
assert "RepartitionExec: partitioning=RoundRobinBatch(4)" in plan
4142

4243

4344
def test_a_context_has_the_information_schema():

0 commit comments

Comments
 (0)