From c5c7dece51eb33152516db39ffa0d4f0c0f15224 Mon Sep 17 00:00:00 2001
From: Xiaoying Wang <xiaoying_wang@sfu.ca>
Date: Tue, 11 Jun 2024 21:02:38 -0400
Subject: [PATCH] fix arrow2 python test

---
 connectorx-python/connectorx/tests/test_arrow.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/connectorx-python/connectorx/tests/test_arrow.py b/connectorx-python/connectorx/tests/test_arrow.py
index 601257f26..dd52371f5 100644
--- 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)
\ No newline at end of file