Fix dtype issues

dask-contrib · Mar 8, 2024 · 3d8a73e · 3d8a73e
1 parent 1121eff
commit 3d8a73e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_acceptance.py b/tests/test_acceptance.py
@@ -51,6 +51,8 @@ def test_reader_all_primitive_types():
     # timestamp differently. This is likely a bug in arrow but the delta result
     # is "more correct".
     expected_ddf["timestamp"] = expected_ddf["timestamp"].astype("datetime64[us]")
+    expected_ddf["timestamp"] = expected_ddf["timestamp"].dt.tz_localize("UTC")
+    expected_ddf["binary"] = expected_ddf["binary"].astype("string[pyarrow]")
     assert_eq(actual_ddf, expected_ddf)