From dd3797f3acabceff447db60929964ed3dd2ccc49 Mon Sep 17 00:00:00 2001 From: Feilong Hou <77430856+FeilongHou@users.noreply.github.com> Date: Wed, 3 Dec 2025 10:05:10 +0800 Subject: [PATCH] test: add timestamptz to more bulk writer case (#46016) Issue #46015 : On branch feature/timestamps Changes to be committed: modified: testcases/test_bulk_insert.py Signed-off-by: Eric Hou Co-authored-by: Eric Hou --- tests/python_client/testcases/test_bulk_insert.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/python_client/testcases/test_bulk_insert.py b/tests/python_client/testcases/test_bulk_insert.py index 4a1620fd40..9396e1a77e 100644 --- a/tests/python_client/testcases/test_bulk_insert.py +++ b/tests/python_client/testcases/test_bulk_insert.py @@ -1596,6 +1596,7 @@ class TestBulkInsert(TestcaseBaseBulkInsert): cf.gen_float_field(name=df.float_field, nullable=nullable), cf.gen_string_field(name=df.string_field, nullable=nullable), cf.gen_json_field(name=df.json_field, nullable=nullable), + cf.gen_timestamptz_field(name=df.timestamp_field, nullable=nullable), cf.gen_array_field(name=df.array_int_field, element_type=DataType.INT64, nullable=nullable), cf.gen_array_field(name=df.array_float_field, element_type=DataType.FLOAT, nullable=nullable), cf.gen_array_field(name=df.array_string_field, element_type=DataType.VARCHAR, max_length=100, nullable=nullable), @@ -1638,6 +1639,7 @@ class TestBulkInsert(TestcaseBaseBulkInsert): df.float_field: 1.0 if not (nullable and random.random() < 0.5) else None, df.string_field: "string" if not (nullable and random.random() < 0.5) else None, df.json_field: json_value[i%len(json_value)] if not (nullable and random.random() < 0.5) else None, + df.timestamp_field: cf.gen_timestamptz_str() if not (nullable and random.random() < 0.5) else None, df.array_int_field: [1, 2] if not (nullable and random.random() < 0.5) else None, df.array_float_field: [1.0, 2.0] if not (nullable and random.random() < 0.5) else None, df.array_string_field: ["string1", "string2"] if not (nullable and random.random() < 0.5) else None, @@ -2044,6 +2046,7 @@ class TestBulkInsert(TestcaseBaseBulkInsert): cf.gen_float_field(name=df.float_field, nullable=nullable), cf.gen_string_field(name=df.string_field, nullable=nullable), cf.gen_json_field(name=df.json_field, nullable=nullable), + cf.gen_timestamptz_field(name=df.timestamp_field, nullable=nullable), cf.gen_array_field(name=df.array_int_field, element_type=DataType.INT64, nullable=nullable), cf.gen_array_field(name=df.array_float_field, element_type=DataType.FLOAT, nullable=nullable), cf.gen_array_field(name=df.array_string_field, element_type=DataType.VARCHAR, max_length=100, nullable=nullable), @@ -2086,6 +2089,7 @@ class TestBulkInsert(TestcaseBaseBulkInsert): df.float_field: 1.0 if not (nullable and random.random() < 0.5) else None, df.string_field: "string" if not (nullable and random.random() < 0.5) else None, df.json_field: json_value[i%len(json_value)] if not (nullable and random.random() < 0.5) else None, + df.timestamp_field: cf.gen_timestamptz_str() if not (nullable and random.random() < 0.5) else None, df.array_int_field: [1, 2] if not (nullable and random.random() < 0.5) else None, df.array_float_field: [1.0, 2.0] if not (nullable and random.random() < 0.5) else None, df.array_string_field: ["string1", "string2"] if not (nullable and random.random() < 0.5) else None,