Fixed Python 2.7 compatibility in trigger tests

agronholm · Dec 22, 2017 · e50b9c6 · e50b9c6
1 parent b7475d3
commit e50b9c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_triggers.py b/tests/test_triggers.py
@@ -599,7 +599,7 @@ def test_jitter(self, trigger, timezone):
         expected = timezone.localize(datetime(2017, 8, 7))
         for _ in range(100):
             next_fire_time = trigger.get_next_fire_time(None, start_time)
-            assert abs(expected.timestamp() - next_fire_time.timestamp()) <= 5
+            assert abs(expected - next_fire_time) <= timedelta(seconds=5)
 
     @pytest.mark.parametrize('jitter', [None, 5], ids=['nojitter', 'jitter'])
     def test_repr(self, trigger, jitter):
@@ -647,7 +647,7 @@ def test_jitter(self, trigger, timezone):
         start_time = expected = timezone.localize(datetime(2017, 8, 6))
         for _ in range(100):
             next_fire_time = trigger.get_next_fire_time(None, start_time)
-            assert abs(expected.timestamp() - next_fire_time.timestamp()) <= 5
+            assert abs(expected - next_fire_time) <= timedelta(seconds=5)
 
     @pytest.mark.parametrize('jitter', [None, 5], ids=['nojitter', 'jitter'])
     def test_repr(self, trigger, jitter):