From d077ca1e7c0303af8863bbd65423c23354389353 Mon Sep 17 00:00:00 2001 From: Fred Drake Date: Wed, 12 Dec 2001 05:38:08 +0000 Subject: Very small test suite for the calendar module, mostly to check a constraint on the return values from isleap(). Also checks firstweekday() and setfirstweekday(). --- Lib/test/test_calendar.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 Lib/test/test_calendar.py diff --git a/Lib/test/test_calendar.py b/Lib/test/test_calendar.py new file mode 100644 index 0000000..1807a59 --- /dev/null +++ b/Lib/test/test_calendar.py @@ -0,0 +1,34 @@ +import calendar +import unittest + +from test_support import run_unittest + + +class CalendarTestCase(unittest.TestCase): + def test_isleap(self): + # Make sure that the return is right for a few years, and + # ensure that the return values are 1 or 0, not just true or + # false (see SF bug #485794). Specific additional tests may + # be appropriate; this tests a single "cycle". + self.assertEqual(calendar.isleap(2000), 1) + self.assertEqual(calendar.isleap(2001), 0) + self.assertEqual(calendar.isleap(2002), 0) + self.assertEqual(calendar.isleap(2003), 0) + + def test_setfirstweekday(self): + self.assertRaises(ValueError, calendar.setfirstweekday, 'flabber') + self.assertRaises(ValueError, calendar.setfirstweekday, -1) + self.assertRaises(ValueError, calendar.setfirstweekday, 200) + orig = calendar.firstweekday() + calendar.setfirstweekday(calendar.SUNDAY) + self.assertEqual(calendar.firstweekday(), calendar.SUNDAY) + calendar.setfirstweekday(calendar.MONDAY) + self.assertEqual(calendar.firstweekday(), calendar.MONDAY) + calendar.setfirstweekday(orig) + + +def test_main(): + run_unittest(CalendarTestCase) + +if __name__ == "__main__": + test_main() -- cgit v0.12