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