From d394be91091939971bdc01dff654ae1463359169 Mon Sep 17 00:00:00 2001 From: Dennis Felsing Date: Fri, 24 Jul 2026 14:05:50 +0200 Subject: [PATCH] tests: Fix date generator Following https://github.com/MaterializeInc/materialize/pull/37838 Noticed in https://buildkite.com/materialize/nightly/builds/17538 --- .../checks/all_checks/datetime_functions.py | 4 ++-- .../materialize/data_ingest/data_type.py | 24 +++++++++---------- 2 files changed, 14 insertions(+), 14 deletions(-) diff --git a/misc/python/materialize/checks/all_checks/datetime_functions.py b/misc/python/materialize/checks/all_checks/datetime_functions.py index 2150d0cfe974e..e0bee3043b83e 100644 --- a/misc/python/materialize/checks/all_checks/datetime_functions.py +++ b/misc/python/materialize/checks/all_checks/datetime_functions.py @@ -75,13 +75,13 @@ def validate(self) -> Testdrive: 3 334 "2024-12-31 23:59:59" "2024-12-31 18:59:59" "2024-12-31 18:59:59" 4 335 "2025-01-01 00:00:00" "2024-12-31 19:00:00" "2024-12-31 19:00:00" - >[version>=2603500] SELECT id, jd::text, jh::text, ji::text FROM datetime_fns_convert1 + >[version>=2603600] SELECT id, jd::text, jh::text, ji::text FROM datetime_fns_convert1 1 "1 mon 5 days" "35 days" "1 mon 5 days" 2 27:00:00 "1 day 03:00:00" "1 day 03:00:00" 3 "1 mon -01:00:00" "1 mon -01:00:00" "29 days 23:00:00" 4 00:00:00 00:00:00 00:00:00 - >[version<2603500] SELECT id, jd::text, jh::text, ji::text FROM datetime_fns_convert1 + >[version<2603600] SELECT id, jd::text, jh::text, ji::text FROM datetime_fns_convert1 1 "1 month 5 days" "35 days" "1 month 5 days" 2 27:00:00 "1 day 03:00:00" "1 day 03:00:00" 3 "1 month -01:00:00" "1 month -01:00:00" "29 days 23:00:00" diff --git a/misc/python/materialize/data_ingest/data_type.py b/misc/python/materialize/data_ingest/data_type.py index 741f09ad6b3ef..a72ab6f87c3dc 100644 --- a/misc/python/materialize/data_ingest/data_type.py +++ b/misc/python/materialize/data_ingest/data_type.py @@ -778,11 +778,11 @@ def random_value( in_query: bool = False, ) -> Any: if rng.randrange(100) == 0: - result = "1-01-01" + result = "0001-01-01" elif rng.randrange(100) == 0: result = "99999-12-31" else: - result = f"{rng.randrange(1, 100000)}-{rng.randrange(1, 13)}-{rng.randrange(1, 29)}" + result = f"{rng.randrange(1, 100000):04}-{rng.randrange(1, 13)}-{rng.randrange(1, 29)}" return f"TIMESTAMP '{result}'" if in_query else str(result) @staticmethod @@ -813,11 +813,11 @@ def random_value( in_query: bool = False, ) -> Any: if rng.randrange(100) == 0: - result = "1-01-01" + result = "0001-01-01" elif rng.randrange(100) == 0: result = "99999-12-31" else: - result = f"{rng.randrange(1, 100000)}-{rng.randrange(1, 13)}-{rng.randrange(1, 29)}" + result = f"{rng.randrange(1, 100000):04}-{rng.randrange(1, 13)}-{rng.randrange(1, 29)}" return f"TIMESTAMPTZ '{result}'" if in_query else str(result) @staticmethod @@ -878,11 +878,11 @@ def random_value( in_query: bool = False, ) -> Any: if rng.randrange(100) == 0: - result = "1-01-01" + result = "0001-01-01" elif rng.randrange(100) == 0: result = "99999-12-31" else: - result = f"{rng.randrange(1, 100000)}-{rng.randrange(1, 13)}-{rng.randrange(1, 29)}" + result = f"{rng.randrange(1, 100000):04}-{rng.randrange(1, 13)}-{rng.randrange(1, 29)}" return f"DATE '{result}'" if in_query else result @staticmethod @@ -1099,8 +1099,8 @@ def random_value( day2 = rng.randrange(1, 29) bounds = rng.choice(["[)", "[]", "()", "(]"]) if in_query: - return f"daterange('{year1}-{month1}-{day1}'::date, '{year2}-{month2}-{day2}'::date, '{bounds}')" - return f"{bounds[0]}{year1}-{month1}-{day1},{year2}-{month2}-{day2}{bounds[1]}" + return f"daterange('{year1:04}-{month1}-{day1}'::date, '{year2:04}-{month2}-{day2}'::date, '{bounds}')" + return f"{bounds[0]}{year1:04}-{month1}-{day1},{year2:04}-{month2}-{day2}{bounds[1]}" @staticmethod def numeric_value(num: int, in_query: bool = False) -> Any: @@ -1134,8 +1134,8 @@ def random_value( day2 = rng.randrange(1, 29) bounds = rng.choice(["[)", "[]", "()", "(]"]) if in_query: - return f"tsrange('{year1}-{month1}-{day1}'::timestamp, '{year2}-{month2}-{day2}'::timestamp, '{bounds}')" - return f"{bounds[0]}{year1}-{month1}-{day1},{year2}-{month2}-{day2}{bounds[1]}" + return f"tsrange('{year1:04}-{month1}-{day1}'::timestamp, '{year2:04}-{month2}-{day2}'::timestamp, '{bounds}')" + return f"{bounds[0]}{year1:04}-{month1}-{day1},{year2:04}-{month2}-{day2}{bounds[1]}" @staticmethod def numeric_value(num: int, in_query: bool = False) -> Any: @@ -1169,8 +1169,8 @@ def random_value( day2 = rng.randrange(1, 29) bounds = rng.choice(["[)", "[]", "()", "(]"]) if in_query: - return f"tstzrange('{year1}-{month1}-{day1}'::timestamptz, '{year2}-{month2}-{day2}'::timestamptz, '{bounds}')" - return f"{bounds[0]}{year1}-{month1}-{day1},{year2}-{month2}-{day2}{bounds[1]}" + return f"tstzrange('{year1:04}-{month1}-{day1}'::timestamptz, '{year2:04}-{month2}-{day2}'::timestamptz, '{bounds}')" + return f"{bounds[0]}{year1:04}-{month1}-{day1},{year2:04}-{month2}-{day2}{bounds[1]}" @staticmethod def numeric_value(num: int, in_query: bool = False) -> Any: