From 8fbab514d68b5633c6b2d6b122b360f4a388568d Mon Sep 17 00:00:00 2001 From: Alex Holyoke Date: Sun, 14 Jul 2024 09:30:16 -0400 Subject: [PATCH] fix: Set cte_follows_insert to True --- sqlalchemy_bigquery/base.py | 1 + .../sqlalchemy_dialect_compliance/test_dialect_compliance.py | 4 ---- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/sqlalchemy_bigquery/base.py b/sqlalchemy_bigquery/base.py index 38d8f5c1..b29ea919 100644 --- a/sqlalchemy_bigquery/base.py +++ b/sqlalchemy_bigquery/base.py @@ -990,6 +990,7 @@ class BigQueryDialect(DefaultDialect): type_compiler = BigQueryTypeCompiler ddl_compiler = BigQueryDDLCompiler execution_ctx_cls = BigQueryExecutionContext + cte_follows_insert = True supports_alter = False supports_comments = True inline_comments = True diff --git a/tests/sqlalchemy_dialect_compliance/test_dialect_compliance.py b/tests/sqlalchemy_dialect_compliance/test_dialect_compliance.py index 57cd9a0d..4af38a84 100644 --- a/tests/sqlalchemy_dialect_compliance/test_dialect_compliance.py +++ b/tests/sqlalchemy_dialect_compliance/test_dialect_compliance.py @@ -537,10 +537,6 @@ def test_round_trip_executemany(self, connection): class CTETest(_CTETest): - @pytest.mark.skip("Can't use CTEs with insert") - def test_insert_from_select_round_trip(self): - pass - @pytest.mark.skip("Recusive CTEs aren't supported.") def test_select_recursive_round_trip(self): pass