Migrated datetime_safe doctests. Thanks to Stephan Jaekel.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13888 bcc190cf-cafb-0310-a4f2-bffc1f526a37

1 files changed, 35 insertions, 37 deletions

diff --git a/tests/regressiontests/utils/datetime_safe.py b/tests/regressiontests/utils/datetime_safe.py
index 6c3e7df359..62b9c7481f 100644
--- a/tests/regressiontests/utils/datetime_safe.py
+++ b/tests/regressiontests/utils/datetime_safe.py
@@ -1,44 +1,42 @@
-"""
->>> from datetime import date as original_date, datetime as original_datetime
->>> from django.utils.datetime_safe import date, datetime
->>> just_safe = (1900, 1, 1)
->>> just_unsafe = (1899, 12, 31, 23, 59, 59)
->>> really_old = (20, 1, 1)
->>> more_recent = (2006, 1, 1)
+from django.test import TestCase
 
->>> original_datetime(*more_recent) == datetime(*more_recent)
-True
->>> original_datetime(*really_old) == datetime(*really_old)
-True
->>> original_date(*more_recent) == date(*more_recent)
-True
->>> original_date(*really_old) == date(*really_old)
-True
+from datetime import date as original_date, datetime as original_datetime
+from django.utils.datetime_safe import date, datetime
 
->>> original_date(*just_safe).strftime('%Y-%m-%d') == date(*just_safe).strftime('%Y-%m-%d')
-True
->>> original_datetime(*just_safe).strftime('%Y-%m-%d') == datetime(*just_safe).strftime('%Y-%m-%d')
-True
+class DatetimeTests(TestCase):
 
->>> date(*just_unsafe[:3]).strftime('%Y-%m-%d (weekday %w)')
-'1899-12-31 (weekday 0)'
->>> date(*just_safe).strftime('%Y-%m-%d (weekday %w)')
-'1900-01-01 (weekday 1)'
+    def setUp(self):
+        self.just_safe = (1900, 1, 1)
+        self.just_unsafe = (1899, 12, 31, 23, 59, 59)
+        self.really_old = (20, 1, 1)
+        self.more_recent = (2006, 1, 1)
 
->>> datetime(*just_unsafe).strftime('%Y-%m-%d %H:%M:%S (weekday %w)')
-'1899-12-31 23:59:59 (weekday 0)'
->>> datetime(*just_safe).strftime('%Y-%m-%d %H:%M:%S (weekday %w)')
-'1900-01-01 00:00:00 (weekday 1)'
+    def test_compare_datetimes(self):
+        self.assertEqual(original_datetime(*self.more_recent), datetime(*self.more_recent))
+        self.assertEqual(original_datetime(*self.really_old), datetime(*self.really_old))
+        self.assertEqual(original_date(*self.more_recent), date(*self.more_recent))
+        self.assertEqual(original_date(*self.really_old), date(*self.really_old))
 
->>> date(*just_safe).strftime('%y')   # %y will error before this date
-'00'
->>> datetime(*just_safe).strftime('%y')
-'00'
+        self.assertEqual(original_date(*self.just_safe).strftime('%Y-%m-%d'), date(*self.just_safe).strftime('%Y-%m-%d'))
+        self.assertEqual(original_datetime(*self.just_safe).strftime('%Y-%m-%d'), datetime(*self.just_safe).strftime('%Y-%m-%d'))
 
->>> date(1850, 8, 2).strftime("%Y/%m/%d was a %A")
-'1850/08/02 was a Friday'
+    def test_safe_strftime(self):
+        self.assertEquals(date(*self.just_unsafe[:3]).strftime('%Y-%m-%d (weekday %w)'), '1899-12-31 (weekday 0)')
+        self.assertEquals(date(*self.just_safe).strftime('%Y-%m-%d (weekday %w)'), '1900-01-01 (weekday 1)')
 
-# Regression for #12524 -- Check that pre-1000AD dates are padded with zeros if necessary
->>> date(1, 1, 1).strftime("%Y/%m/%d was a %A")
-'0001/01/01 was a Monday'
-"""
+        self.assertEquals(datetime(*self.just_unsafe).strftime('%Y-%m-%d %H:%M:%S (weekday %w)'), '1899-12-31 23:59:59 (weekday 0)')
+        self.assertEquals(datetime(*self.just_safe).strftime('%Y-%m-%d %H:%M:%S (weekday %w)'), '1900-01-01 00:00:00 (weekday 1)')
+
+        # %y will error before this date
+        self.assertEquals(date(*self.just_safe).strftime('%y'), '00')
+        self.assertEquals(datetime(*self.just_safe).strftime('%y'), '00')
+
+        self.assertEquals(date(1850, 8, 2).strftime("%Y/%m/%d was a %A"), '1850/08/02 was a Friday')
+
+    def test_zero_padding(self):
+        """
+        Regression for #12524
+
+        Check that pre-1000AD dates are padded with zeros if necessary
+        """
+        self.assertEquals(date(1, 1, 1).strftime("%Y/%m/%d was a %A"), '0001/01/01 was a Monday')