From 9f8b1b5301ecfedf669d1dd2ad3227633001c2f7 Mon Sep 17 00:00:00 2001 From: John Bodley <4567245+john-bodley@users.noreply.github.com> Date: Sat, 9 Oct 2021 15:18:59 -0700 Subject: [PATCH] Update druid_tests.py --- tests/integration_tests/db_engine_specs/druid_tests.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration_tests/db_engine_specs/druid_tests.py b/tests/integration_tests/db_engine_specs/druid_tests.py index 4fc7e27b3483e..22fcfee3ab28d 100644 --- a/tests/integration_tests/db_engine_specs/druid_tests.py +++ b/tests/integration_tests/db_engine_specs/druid_tests.py @@ -52,8 +52,8 @@ def test_timegrain_expressions(self): test_cases = { "PT1S": f"TIME_FLOOR({col}, 'PT1S')", "PT5M": f"TIME_FLOOR({col}, 'PT5M')", - "P1W/1970-01-03T00:00:00Z": "TIME_SHIFT(TIME_FLOOR(TIME_SHIFT({col}, 'P1D', 1), 'P1W'), 'P1D', 5)", - "1969-12-28T00:00:00Z/P1W": "TIME_SHIFT(TIME_FLOOR(TIME_SHIFT({col}, 'P1D', 1), 'P1W'), 'P1D', -1)", + f"P1W/1970-01-03T00:00:00Z": "TIME_SHIFT(TIME_FLOOR(TIME_SHIFT({col}, 'P1D', 1), 'P1W'), 'P1D', 5)", + f"1969-12-28T00:00:00Z/P1W": "TIME_SHIFT(TIME_FLOOR(TIME_SHIFT({col}, 'P1D', 1), 'P1W'), 'P1D', -1)", } for grain, expected in test_cases.items(): actual = DruidEngineSpec.get_timestamp_expr(