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"]
)