py3 fix

taichino · Apr 13, 2012 · cd4ec6b · cd4ec6b
1 parent a2a83a2
commit cd4ec6b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/croniter/tests/test_croniter.py b/src/croniter/tests/test_croniter.py
@@ -394,8 +394,8 @@ def test_bug34(self):
         itr = croniter('* * 31 2 *', base)
         try:
             itr.get_next(datetime)
-        except CroniterBadDateError, ex:
-            self.assertEqual(ex.message,
+        except (CroniterBadDateError,) as ex:
+            self.assertEqual("{0}".format(ex),
                              'failed to find next date')
 
     def testBug57(self):
@@ -512,7 +512,8 @@ def assertScheduleTimezone(self, callback, expected_schedule):
         for expected_date, expected_offset in expected_schedule:
             d = callback()
             self.assertEqual(expected_date, d.replace(tzinfo=None))
-            self.assertEqual(expected_offset, croniter._timedelta_to_seconds(d.utcoffset()))
+            self.assertEqual(expected_offset,
+                             croniter._timedelta_to_seconds(d.utcoffset()))
 
     def testTimezoneWinterTime(self):
         tz = pytz.timezone('Europe/Athens')