Merge branch 'main' into fix-table-reflection

Author: JacobHayes

samples/snippets/requirements.txt

@@ -1,5 +1,5 @@
 alembic==1.13.1
-certifi==2024.6.2
+certifi==2024.7.4
 charset-normalizer==3.3.2
 geoalchemy2==0.15.1
 google-api-core[grpc]==2.19.0

sqlalchemy_bigquery/base.py

@@ -579,6 +579,9 @@ def visit_regexp_match_op_binary(self, binary, operator, **kw):
     def visit_not_regexp_match_op_binary(self, binary, operator, **kw):
         return "NOT %s" % self.visit_regexp_match_op_binary(binary, operator, **kw)
 
+    def visit_mod_binary(self, binary, operator, **kw):
+        return f"MOD({self.process(binary.left, **kw)}, {self.process(binary.right, **kw)})"
+
 
 class BigQueryTypeCompiler(GenericTypeCompiler):
     def visit_INTEGER(self, type_, **kw):
@@ -985,6 +988,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

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

tests/unit/test_select.py

@@ -406,3 +406,16 @@ def test_visit_not_regexp_match_op_binary(faux_conn):
     expected = "NOT REGEXP_CONTAINS(`table`.`foo`, %(foo_1:STRING)s)"
 
     assert result == expected
+
+
+def test_visit_mod_binary(faux_conn):
+    table = setup_table(
+        faux_conn,
+        "table",
+        sqlalchemy.Column("foo", sqlalchemy.Integer),
+    )
+    sql_statement = table.c.foo % 2
+    result = sql_statement.compile(faux_conn).string
+    expected = "MOD(`table`.`foo`, %(foo_1:INT64)s)"
+
+    assert result == expected