diff --git a/tests/test_acceptance.py b/tests/test_acceptance.py index 17e99ff..ecb460a 100644 --- 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)