fix: Set cte_follows_insert to True (#1095)

googleapis · Jul 18, 2024 · 9e0b117 · 9e0b117
1 parent f5fb1a2
commit 9e0b117
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git 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
@@ -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