Skip to content

Commit

Permalink
update tests to expect lowercase sum in query plans
Browse files Browse the repository at this point in the history
This was changed upstream.

Ref: apache/datafusion#10831
  • Loading branch information
Michael-J-Ward committed Jul 24, 2024
1 parent d34b0d7 commit 3a2784d
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions python/datafusion/tests/test_dataframe.py
Original file line number Diff line number Diff line change
Expand Up @@ -434,13 +434,13 @@ def test_explain(df):
def test_logical_plan(aggregate_df):
plan = aggregate_df.logical_plan()

expected = "Projection: test.c1, SUM(test.c2)"
expected = "Projection: test.c1, sum(test.c2)"

assert expected == plan.display()

expected = (
"Projection: test.c1, SUM(test.c2)\n"
" Aggregate: groupBy=[[test.c1]], aggr=[[SUM(test.c2)]]\n"
"Projection: test.c1, sum(test.c2)\n"
" Aggregate: groupBy=[[test.c1]], aggr=[[sum(test.c2)]]\n"
" TableScan: test"
)

Expand All @@ -450,12 +450,12 @@ def test_logical_plan(aggregate_df):
def test_optimized_logical_plan(aggregate_df):
plan = aggregate_df.optimized_logical_plan()

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

assert expected == plan.display()

expected = (
"Aggregate: groupBy=[[test.c1]], aggr=[[SUM(test.c2)]]\n"
"Aggregate: groupBy=[[test.c1]], aggr=[[sum(test.c2)]]\n"
" TableScan: test projection=[c1, c2]"
)

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

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

assert expected == plan.display()
Expand Down

0 comments on commit 3a2784d

Please sign in to comment.