casting datetime to datetime64[us]

coiled · Jun 25, 2024 · 498dbec · 498dbec
1 parent d071808
commit 498dbec
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/dask_bigquery/tests/test_core.py b/dask_bigquery/tests/test_core.py
@@ -34,8 +34,15 @@ def df():
         }
         for i in range(10)
     ]
+    df = pd.DataFrame(records)
 
-    yield pd.DataFrame(records)
+    df["timestamp"] = (
+        df["timestamp"]
+        .dt.tz_convert("UTC")
+        .dt.tz_localize(None)
+        .astype("datetime64[us]")
+    )
+    yield df
 
 
 @pytest.fixture(scope="module")