dbcli/sqlglot
1
0
Fork 0
Code Issues Activity

Adding upstream version 26.11.1.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-03-23 10:41:59 +01:00
parent 5b8e67f8b8
commit aa70b5e889
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
52 changed files with 14319 additions and 14244 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tests/test_optimizer.py
View file

@ -180,13 +180,13 @@ class TestOptimizer(unittest.TestCase):
actual,
)
if string_to_bool(execute):
with self.subTest(f"(execute) {title}"):
df1 = self.conn.execute(
sqlglot.transpile(sql, read=dialect, write="duckdb")[0]
).df()
df2 = self.conn.execute(optimized.sql(dialect="duckdb")).df()
assert_frame_equal(df1, df2)
if string_to_bool(execute):
with self.subTest(f"(execute) {title}"):
df1 = self.conn.execute(
sqlglot.transpile(sql, read=dialect, write="duckdb")[0]
).df()
df2 = self.conn.execute(optimized.sql(dialect="duckdb")).df()
assert_frame_equal(df1, df2)
@patch("sqlglot.generator.logger")
def test_optimize(self, logger):
@ -659,6 +659,15 @@ SELECT :with,WITH :expressions,CTE :this,UNION :this,SELECT :expressions,1,:expr
"WITH data AS (SELECT 1 AS id) SELECT FUNC(data.id) AS id FROM data GROUP BY FUNC(data.id)",
)
sql = "SELECT x.a, max(x.b) as x FROM x AS x GROUP BY 1 HAVING x > 1"
self.assertEqual(
optimizer.qualify_columns.qualify_columns(
parse_one(sql, dialect="bigquery"),
schema=MappingSchema(schema=unused_schema, dialect="bigquery"),
).sql(),
"SELECT x.a AS a, MAX(x.b) AS x FROM x AS x GROUP BY 1 HAVING x > 1",
)
def test_optimize_joins(self):
self.check_file(
"optimize_joins",
@ -1425,6 +1434,16 @@ FROM READ_CSV('tests/fixtures/optimizer/tpc-h/nation.csv.gz', 'delimiter', '|')
"""
self.assertEqual(optimizer.optimize(sql).selects[0].type.this, exp.DataType.Type.VARCHAR)
def test_udtf_annotation(self):
table_udtf = parse_one(
"SELECT * FROM TABLE(GENERATOR(ROWCOUNT => 100000))",
read="snowflake",
)
self.assertEqual(
annotate_types(table_udtf, dialect="snowflake").sql("snowflake"),
"SELECT * FROM TABLE(GENERATOR(ROWCOUNT => 100000))",
)
def test_recursive_cte(self):
query = parse_one(
"""