diff options
Diffstat (limited to 'tests/test_executor.py')
-rw-r--r-- | tests/test_executor.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/test_executor.py b/tests/test_executor.py index 013ff34..c455f3a 100644 --- a/tests/test_executor.py +++ b/tests/test_executor.py @@ -401,6 +401,16 @@ class TestExecutor(unittest.TestCase): ], ) + table1_view = exp.Select().select("id", "sub_type").from_("table1").subquery() + select_from_sub_query = exp.Select().select("id AS id_alias", "sub_type").from_(table1_view) + expression = exp.Select().select("*").from_("cte1").with_("cte1", as_=select_from_sub_query) + + schema = {"table1": {"id": "str", "sub_type": "str"}} + executed = execute(expression, tables={t: [] for t in schema}, schema=schema) + + self.assertEqual(executed.rows, []) + self.assertEqual(executed.columns, ("id_alias", "sub_type")) + def test_correlated_count(self): tables = { "parts": [{"pnum": 0, "qoh": 1}], |