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,