Update test after flattening

sfc-gh-yixie · sfc-gh-yixie · commit 1d3ad208a404 · 2025-10-28T10:53:52.000-07:00
diff --git a/tests/integ/test_query_line_intervals.py b/tests/integ/test_query_line_intervals.py
@@ -73,7 +73,7 @@ def generate_test_data(session, sql_simplifier_enabled):
             lambda data: data["df1"].filter(data["df1"].value > 150),
             True,
             {
-                8: 'SELECT $1 AS "_1", $2 AS "_2", $3 AS "_3" FROM VALUES (1 :: INT, \'A\' :: STRING, 100 :: INT), (2 :: INT, \'B\' :: STRING, 200 :: INT)',
+                8: """SELECT "_1" AS "ID", "_2" AS "NAME", "_3" AS "VALUE" FROM (SELECT $1 AS "_1", $2 AS "_2", $3 AS "_3" FROM VALUES (1 :: INT, 'A' :: STRING, 100 :: INT), (2 :: INT, 'B' :: STRING, 200 :: INT)) WHERE ("VALUE" > 150)""",
             },
         ),
         (

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def generate_test_data(session, sql_simplifier_enabled):`
`73`	`73`	`lambda data: data["df1"].filter(data["df1"].value > 150),`
`74`	`74`	`True,`
`75`	`75`	`{`
`76`		`- 8: 'SELECT $1 AS "_1", $2 AS "_2", $3 AS "_3" FROM VALUES (1 :: INT, \'A\' :: STRING, 100 :: INT), (2 :: INT, \'B\' :: STRING, 200 :: INT)',`
	`76`	`+ 8: """SELECT "_1" AS "ID", "_2" AS "NAME", "_3" AS "VALUE" FROM (SELECT $1 AS "_1", $2 AS "_2", $3 AS "_3" FROM VALUES (1 :: INT, 'A' :: STRING, 100 :: INT), (2 :: INT, 'B' :: STRING, 200 :: INT)) WHERE ("VALUE" > 150)""",`
`77`	`77`	`},`
`78`	`78`	`),`
`79`	`79`	`(`