mirror of https://github.com/python/cpython
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import calendar
|
|
import unittest
|
|
|
|
from test.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_enumerateweekdays(self):
|
|
self.assertRaises(IndexError, calendar.day_abbr.__getitem__, -10)
|
|
self.assertRaises(IndexError, calendar.day_name.__getitem__, 10)
|
|
self.assertEqual(len([d for d in calendar.day_abbr]), 7)
|
|
|
|
def test_days(self):
|
|
for attr in "day_name", "day_abbr":
|
|
value = getattr(calendar, attr)
|
|
self.assertEqual(len(value), 7)
|
|
self.assertEqual(len(value[:]), 7)
|
|
# ensure they're all unique
|
|
d = {}
|
|
for v in value:
|
|
d[v] = 1
|
|
self.assertEqual(len(d), 7)
|
|
|
|
def test_months(self):
|
|
for attr in "month_name", "month_abbr":
|
|
value = getattr(calendar, attr)
|
|
self.assertEqual(len(value), 13)
|
|
self.assertEqual(len(value[:]), 13)
|
|
self.assertEqual(value[0], "")
|
|
# ensure they're all unique
|
|
d = {}
|
|
for v in value:
|
|
d[v] = 1
|
|
self.assertEqual(len(d), 13)
|
|
|
|
def test_main():
|
|
run_unittest(CalendarTestCase)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|