diff --git a/tests/integration_tests/db_engine_specs/bigquery_tests.py b/tests/integration_tests/db_engine_specs/bigquery_tests.py index 5d0daf855e7b0..32e3d1f2067e5 100644 --- 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" )