dbcli/sqlglot
1
0
Fork 0
Code Issues Activity

Adding upstream version 20.3.0.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-02-13 21:17:33 +01:00
parent 5bd573dda1
commit fd9de5e4cb
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
132 changed files with 55125 additions and 51576 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/test_optimizer.py
View file

@ -109,7 +109,9 @@ class TestOptimizer(unittest.TestCase):
},
}
def check_file(self, file, func, pretty=False, execute=False, set_dialect=False, **kwargs):
def check_file(
self, file, func, pretty=False, execute=False, set_dialect=False, only=None, **kwargs
):
with ProcessPoolExecutor() as pool:
results = {}
@ -117,6 +119,8 @@ class TestOptimizer(unittest.TestCase):
load_sql_fixture_pairs(f"optimizer/{file}.sql"), start=1
):
title = meta.get("title") or f"{i}, {sql}"
if only and title != only:
continue
dialect = meta.get("dialect")
leave_tables_isolated = meta.get("leave_tables_isolated")
@ -137,13 +141,14 @@ class TestOptimizer(unittest.TestCase):
)
for future in as_completed(results):
optimized = future.result()
sql, title, expected, dialect, execute = results[future]
with self.subTest(title):
optimized = future.result()
actual = optimized.sql(pretty=pretty, dialect=dialect)
self.assertEqual(
expected,
optimized.sql(pretty=pretty, dialect=dialect),
actual,
)
if string_to_bool(execute):
@ -309,7 +314,7 @@ class TestOptimizer(unittest.TestCase):
self.check_file("pushdown_projections", pushdown_projections, schema=self.schema)
def test_simplify(self):
self.check_file("simplify", simplify)
self.check_file("simplify", simplify, set_dialect=True)
expression = parse_one("TRUE AND TRUE AND TRUE")
self.assertEqual(exp.true(), optimizer.simplify.simplify(expression))