Test calendar.monthrange.
Closes issue 9342. Thanks John Chandler for the patch.
diff --git a/Lib/test/test_calendar.py b/Lib/test/test_calendar.py
index b7ce475..b936acb 100644
--- a/Lib/test/test_calendar.py
+++ b/Lib/test/test_calendar.py
@@ -389,6 +389,34 @@
tuple = time.gmtime(secs)
self.assertEqual(secs, calendar.timegm(tuple))
+class MonthRangeTestCase(unittest.TestCase):
+ def test_january(self):
+ # Tests valid lower boundary case.
+ self.assertEqual(calendar.monthrange(2004,1), (3,31))
+
+ def test_february_leap(self):
+ # Tests February during leap year.
+ self.assertEqual(calendar.monthrange(2004,2), (6,29))
+
+ def test_february_nonleap(self):
+ # Tests February in non-leap year.
+ self.assertEqual(calendar.monthrange(2010,2), (0,28))
+
+ def test_december(self):
+ # Tests valid upper boundary case.
+ self.assertEqual(calendar.monthrange(2004,12), (2,31))
+
+ def test_zeroth_month(self):
+ # Tests low invalid boundary case.
+ with self.assertRaises(calendar.IllegalMonthError):
+ calendar.monthrange(2004, 0)
+
+ def test_thirteenth_month(self):
+ # Tests high invalid boundary case.
+ with self.assertRaises(calendar.IllegalMonthError):
+ calendar.monthrange(2004, 13)
+
+
def test_main():
support.run_unittest(
OutputTestCase,
@@ -396,6 +424,7 @@
MondayTestCase,
SundayTestCase,
TimegmTestCase,
+ MonthRangeTestCase,
)