fix UT

apache · Sep 19, 2022 · 9883d9f · 9883d9f
1 parent 8750ec8
commit 9883d9f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/integration_tests/db_engine_specs/bigquery_tests.py b/tests/integration_tests/db_engine_specs/bigquery_tests.py
@@ -77,6 +77,7 @@ def test_timegrain_expressions(self):
             "TIMESTAMP": "TIMESTAMP_TRUNC(temporal, HOUR)",
         }
         for type_, expected in test_cases.items():
+            col.type = type_
             actual = BigQueryEngineSpec.get_timestamp_expr(
                 col=col, pdf=None, time_grain="PT1H"
             )
@@ -99,6 +100,7 @@ def test_custom_minute_timegrain_expressions(self):
             ") AS TIMESTAMP)",
         }
         for type_, expected in test_cases.items():
+            col.type = type_
             actual = BigQueryEngineSpec.get_timestamp_expr(
                 col=col, pdf=None, time_grain="PT5M"
             )