migrate away from unittest and adopt pytest for test_utils.py

dateutil · Nov 24, 2019 · 38f17ae · 38f17ae
1 parent 376b330
commit 38f17ae
Showing 1 changed file with 30 additions and 31 deletions.
diff --git a/dateutil/test/test_utils.py b/dateutil/test/test_utils.py
@@ -2,8 +2,6 @@
 from __future__ import unicode_literals
 from datetime import timedelta, datetime
 
-import unittest
-
 from dateutil import tz
 from dateutil import utils
 from dateutil.tz import UTC
@@ -14,40 +12,41 @@
 NYC = tz.gettz("America/New_York")
 
 
-class UtilsTest(unittest.TestCase):
-    @freeze_time(datetime(2014, 12, 15, 1, 21, 33, 4003))
-    def testToday(self):
-        self.assertEqual(utils.today(), datetime(2014, 12, 15, 0, 0, 0))
+@freeze_time(datetime(2014, 12, 15, 1, 21, 33, 4003))
+def test_utils_today():
+    assert utils.today() == datetime(2014, 12, 15, 0, 0, 0)
+
+
+@freeze_time(datetime(2014, 12, 15, 12), tz_offset=5)
+def test_utils_today_tz_info():
+    assert utils.today(NYC) == datetime(2014, 12, 15, 0, 0, 0, tzinfo=NYC)
+
+
+@freeze_time(datetime(2014, 12, 15, 23), tz_offset=5)
+def test_utils_today_tz_info_different_day():
+    assert utils.today(UTC) == datetime(2014, 12, 16, 0, 0, 0, tzinfo=UTC)
+
+
+def test_utils_default_tz_info_naive():
+    dt = datetime(2014, 9, 14, 9, 30)
+    assert utils.default_tzinfo(dt, NYC).tzinfo is NYC
 
-    @freeze_time(datetime(2014, 12, 15, 12), tz_offset=5)
-    def testTodayTzInfo(self):
-        self.assertEqual(utils.today(NYC),
-                         datetime(2014, 12, 15, 0, 0, 0, tzinfo=NYC))
 
-    @freeze_time(datetime(2014, 12, 15, 23), tz_offset=5)
-    def testTodayTzInfoDifferentDay(self):
-        self.assertEqual(utils.today(UTC),
-                         datetime(2014, 12, 16, 0, 0, 0, tzinfo=UTC))
+def test_utils_default_tz_info_aware():
+    dt = datetime(2014, 9, 14, 9, 30, tzinfo=UTC)
+    assert utils.default_tzinfo(dt, NYC).tzinfo is UTC
 
-    def testDefaultTZInfoNaive(self):
-        dt = datetime(2014, 9, 14, 9, 30)
-        self.assertIs(utils.default_tzinfo(dt, NYC).tzinfo,
-                      NYC)
 
-    def testDefaultTZInfoAware(self):
-        dt = datetime(2014, 9, 14, 9, 30, tzinfo=UTC)
-        self.assertIs(utils.default_tzinfo(dt, NYC).tzinfo,
-                      UTC)
+def test_utils_within_delta():
+    d1 = datetime(2016, 1, 1, 12, 14, 1, 9)
+    d2 = d1.replace(microsecond=15)
 
-    def testWithinDelta(self):
-        d1 = datetime(2016, 1, 1, 12, 14, 1, 9)
-        d2 = d1.replace(microsecond=15)
+    assert within_delta(d1, d2, timedelta(seconds=1))
+    assert not within_delta(d1, d2, timedelta(microseconds=1))
 
-        self.assertTrue(within_delta(d1, d2, timedelta(seconds=1)))
-        self.assertFalse(within_delta(d1, d2, timedelta(microseconds=1)))
 
-    def testWithinDeltaWithNegativeDelta(self):
-        d1 = datetime(2016, 1, 1)
-        d2 = datetime(2015, 12, 31)
+def test_utils_within_delta_with_negative_delta():
+    d1 = datetime(2016, 1, 1)
+    d2 = datetime(2015, 12, 31)
 
-        self.assertTrue(within_delta(d2, d1, timedelta(days=-1)))
+    assert within_delta(d2, d1, timedelta(days=-1))