File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed
Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments