Blob Blame History Raw
From ef65e71446eb0b84df828cd129b639c68ffcbef4 Mon Sep 17 00:00:00 2001
From: scottbelden <scottabelden@gmail.com>
Date: Mon, 25 Oct 2021 17:47:51 -0400
Subject: [PATCH] make test more robust

---
 developer_requirements.txt  |  1 -
 tests/test_logical_types.py | 12 ++++--------
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/developer_requirements.txt b/developer_requirements.txt
index 23d6fbf4..6f7216f2 100644
--- a/developer_requirements.txt
+++ b/developer_requirements.txt
@@ -9,7 +9,6 @@ wheel
 twine
 coverage
 codecov
-python-dateutil
 mypy; implementation_name != "pypy"
 black; implementation_name != "pypy"
 
diff --git a/tests/test_logical_types.py b/tests/test_logical_types.py
index 93ac56da..db7583e9 100644
--- a/tests/test_logical_types.py
+++ b/tests/test_logical_types.py
@@ -9,7 +9,6 @@
 import datetime
 import sys
 import os
-from dateutil.tz import tzlocal
 from datetime import timezone, timedelta
 import numpy as np
 
@@ -147,9 +146,6 @@ def test_ancient_datetime():
         "type": "record",
     }
 
-    my_epoch = datetime.datetime(1970, 1, 1, tzinfo=tzlocal())
-    diff = my_epoch - datetime.datetime(1970, 1, 1, tzinfo=timezone.utc)
-
     data1 = {
         "timestamp-millis": datetime.datetime(1960, 1, 1),
         "timestamp-micros": datetime.datetime(1960, 1, 1),
@@ -157,11 +153,11 @@ def test_ancient_datetime():
     binary = serialize(schema_datetime, data1)
     data2 = deserialize(schema_datetime, binary)
 
-    assert (data1["timestamp-millis"] + diff) == data2["timestamp-millis"].replace(
-        tzinfo=None
+    assert_naive_datetime_equal_to_tz_datetime(
+        data1["timestamp-millis"], data2["timestamp-millis"]
     )
-    assert (data1["timestamp-micros"] + diff) == data2["timestamp-micros"].replace(
-        tzinfo=None
+    assert_naive_datetime_equal_to_tz_datetime(
+        data1["timestamp-micros"], data2["timestamp-micros"]
     )