fix: Support partitioning by DATE or Date

Author: bnaul

sqlalchemy_bigquery/base.py

@@ -838,7 +838,7 @@ def _process_time_partitioning(
             field = time_partitioning.field
             if isinstance(
                 table.columns[time_partitioning.field].type,
-                sqlalchemy.sql.sqltypes.DATE,
+                (sqlalchemy.sql.sqltypes.DATE, sqlalchemy.sql.sqltypes.Date),
             ):
                 return f"PARTITION BY {field}"
             elif isinstance(

tests/unit/test_table_options.py

@@ -193,14 +193,15 @@ def test_table_time_partitioning_with_timestamp_dialect_option(faux_conn):
     )
 
 
-def test_table_time_partitioning_with_date_dialect_option(faux_conn):
+@pytest.mark.parametrize("date_type", [sqlalchemy.DATE, sqlalchemy.Date])
+def test_table_time_partitioning_with_date_dialect_option(faux_conn, date_type):
     # expect table creation to fail as SQLite does not support partitioned tables
     with pytest.raises(sqlite3.OperationalError):
         setup_table(
             faux_conn,
             "some_table_2",
             sqlalchemy.Column("id", sqlalchemy.Integer),
-            sqlalchemy.Column("createdAt", sqlalchemy.DATE),
+            sqlalchemy.Column("createdAt", date_type),
             bigquery_time_partitioning=TimePartitioning(field="createdAt"),
         )