SNOW-1843690: fix daily precommit test failiure (#2706)

snowflakedb · Dec 4, 2024 · a5ab0b6 · a5ab0b6
1 parent b76912f
commit a5ab0b6
Showing 1 changed file with 10 additions and 14 deletions.
diff --git a/tests/integ/scala/test_dataframe_copy_into.py b/tests/integ/scala/test_dataframe_copy_into.py
@@ -30,9 +30,6 @@
     TestFiles,
     Utils,
     iceberg_supported,
-    add_to_time,
-    local_to_utc_offset_in_hours,
-    running_on_public_ci,
 )
 
 test_file_csv = "testCSV.csv"
@@ -979,18 +976,17 @@ def test_copy_non_csv_transformation(
                     S="string",
                     C="a",
                     D=datetime.date(2022, 4, 1),
-                    T=add_to_time(
-                        datetime.time(4, 11, 11),
-                        datetime.timedelta(hours=-local_to_utc_offset_in_hours()),
+                    T=datetime.time(11, 11, 11),
+                    TS_NTZ=datetime.datetime(
+                        2022, 4, 1, 11, 11, 11, tzinfo=datetime.timezone.utc
                     )
-                    if not running_on_public_ci()
-                    else datetime.time(11, 11, 11),
-                    TS_NTZ=datetime.datetime(2022, 4, 1, 4, 11, 11)
-                    if not running_on_public_ci()
-                    else datetime.datetime(2022, 4, 1, 11, 11, 11),
-                    TS=datetime.datetime(2022, 4, 1, 4, 11, 11)
-                    if not running_on_public_ci()
-                    else datetime.datetime(2022, 4, 1, 11, 11, 11),
+                    .astimezone()
+                    .replace(tzinfo=None),
+                    TS=datetime.datetime(
+                        2022, 4, 1, 11, 11, 11, tzinfo=datetime.timezone.utc
+                    )
+                    .astimezone()
+                    .replace(tzinfo=None),
                     V='{"key":"value"}',
                 )
             ],