test_schedule.py: fix mock now() and tests with a timezone

Signed-off-by: Steve Arnold <nerdboy@gentoo.org>

Author: sarnold

test_schedule.py

@@ -36,9 +36,10 @@ def today(cls):
                 return cls(self.year, self.month, self.day)
 
             @classmethod
-            def now(cls):
+            def now(cls, tz=None):
                 return cls(self.year, self.month, self.day,
-                           self.hour, self.minute)
+                           self.hour, self.minute).replace(tzinfo=tz)
+
         self.original_datetime = datetime.datetime
         datetime.datetime = MockDate
 
@@ -258,7 +259,8 @@ def test_next_run_property(self):
             every().hour.do(hourly_job)
             assert len(schedule.jobs) == 2
             # Make sure the hourly job is first
-            assert schedule.next_run() == original_datetime(2010, 1, 6, 14, 16)
+            assert schedule.next_run() == original_datetime(2010, 1, 6, 14, 16,
+                                                            tzinfo=datetime.timezone.utc)
             assert schedule.idle_seconds() == 60 * 60
 
     def test_cancel_job(self):