fix arrow2 python test

sfu-db · Jun 12, 2024 · c5c7dec · c5c7dec
1 parent 0065765
commit c5c7dec
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/connectorx-python/connectorx/tests/test_arrow.py b/connectorx-python/connectorx/tests/test_arrow.py
@@ -92,7 +92,7 @@ def test_arrow2_type(postgres_url: str) -> None:
                     "2038-01-18 23:59:59",
                     None,
                 ],
-                dtype="datetime64[ns]",
+                dtype="datetime64[us]",
             ),
             "test_timestamptz": pd.Series(
                 [
@@ -101,7 +101,7 @@ def test_arrow2_type(postgres_url: str) -> None:
                     "2038-01-18 15:59:59+00:00",
                     None,
                 ],
-                dtype="datetime64[ns, UTC]",
+                dtype="datetime64[us, UTC]",
             ),
             "test_int16": pd.Series([0, 1, 2, 3], dtype="int32"),
             "test_int64": pd.Series(
@@ -190,4 +190,4 @@ def test_arrow2_type(postgres_url: str) -> None:
 
         },
     )
-    assert_frame_equal(df, expected, check_names=True)
+    assert_frame_equal(df, expected, check_names=True)