Skip to content

Commit aaeb854

Browse files
pr feedback
1 parent 2e98e81 commit aaeb854

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/core/test_context.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2782,7 +2782,7 @@ def test_defaults_pre_post_statements(tmp_path: Path):
27822782

27832783
# Verify the statements contain the expected SQL
27842784
assert model.pre_statements[0].sql() == "SET memory_limit = '10GB'"
2785-
assert model.pre_statements[0].sql()
2785+
assert model.render_pre_statements()[0].sql() == "SET \"memory_limit\" = '10GB'"
27862786
assert model.pre_statements[1].sql() == "SET threads = @var1"
27872787
assert model.render_pre_statements()[1].sql() == 'SET "threads" = 4'
27882788

@@ -2815,6 +2815,7 @@ def test_defaults_pre_post_statements(tmp_path: Path):
28152815
# Verify updated statements
28162816
assert len(new_model.pre_statements) == 1
28172817
assert new_model.pre_statements[0].sql() == "SET memory_limit = '5GB'"
2818+
assert new_model.render_pre_statements()[0].sql() == "SET \"memory_limit\" = '5GB'"
28182819

28192820
# Verify the change was detected by the plan
28202821
assert len(updated_plan.directly_modified) == 1

0 commit comments

Comments
 (0)