Skip to content

Commit 11e601f

Browse files
update tests to expect lowercase sum in query plans
This was changed upstream. Ref: apache/datafusion#10831
1 parent 6539c0c commit 11e601f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

python/datafusion/tests/test_dataframe.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -434,13 +434,13 @@ def test_explain(df):
434434
def test_logical_plan(aggregate_df):
435435
plan = aggregate_df.logical_plan()
436436

437-
expected = "Projection: test.c1, SUM(test.c2)"
437+
expected = "Projection: test.c1, sum(test.c2)"
438438

439439
assert expected == plan.display()
440440

441441
expected = (
442-
"Projection: test.c1, SUM(test.c2)\n"
443-
" Aggregate: groupBy=[[test.c1]], aggr=[[SUM(test.c2)]]\n"
442+
"Projection: test.c1, sum(test.c2)\n"
443+
" Aggregate: groupBy=[[test.c1]], aggr=[[sum(test.c2)]]\n"
444444
" TableScan: test"
445445
)
446446

@@ -450,12 +450,12 @@ def test_logical_plan(aggregate_df):
450450
def test_optimized_logical_plan(aggregate_df):
451451
plan = aggregate_df.optimized_logical_plan()
452452

453-
expected = "Aggregate: groupBy=[[test.c1]], aggr=[[SUM(test.c2)]]"
453+
expected = "Aggregate: groupBy=[[test.c1]], aggr=[[sum(test.c2)]]"
454454

455455
assert expected == plan.display()
456456

457457
expected = (
458-
"Aggregate: groupBy=[[test.c1]], aggr=[[SUM(test.c2)]]\n"
458+
"Aggregate: groupBy=[[test.c1]], aggr=[[sum(test.c2)]]\n"
459459
" TableScan: test projection=[c1, c2]"
460460
)
461461

@@ -466,7 +466,7 @@ def test_execution_plan(aggregate_df):
466466
plan = aggregate_df.execution_plan()
467467

468468
expected = (
469-
"AggregateExec: mode=FinalPartitioned, gby=[c1@0 as c1], aggr=[SUM(test.c2)]\n" # noqa: E501
469+
"AggregateExec: mode=FinalPartitioned, gby=[c1@0 as c1], aggr=[sum(test.c2)]\n" # noqa: E501
470470
)
471471

472472
assert expected == plan.display()

0 commit comments

Comments
 (0)