Skip to content

Commit b58417e

Browse files
authored
SNOW-3590670: fix test failure (#4255)
1 parent 4afc47d commit b58417e

1 file changed

Lines changed: 8 additions & 5 deletions

File tree

tests/integ/test_df_aggregate.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1421,11 +1421,14 @@ def test_copy_preserves_agg_state(session):
14211421
copied._agg_base_plan
14221422
and copied._agg_base_plan == agg_sorted._agg_base_plan
14231423
)
1424-
assert (
1425-
copied._agg_base_select_statement
1426-
and copied._agg_base_select_statement
1427-
is agg_sorted._agg_base_select_statement
1428-
)
1424+
# _agg_base_select_statement is only populated when the SQL simplifier
1425+
# is enabled; it is None (and legitimately so) when disabled.
1426+
if session.sql_simplifier_enabled:
1427+
assert (
1428+
copied._agg_base_select_statement
1429+
and copied._agg_base_select_statement
1430+
is agg_sorted._agg_base_select_statement
1431+
)
14291432
assert (
14301433
copied._pending_order_bys
14311434
and copied._pending_order_bys == agg_sorted._pending_order_bys

0 commit comments

Comments
 (0)