def test_invalid_weekdays(self): with self.assertRaises(InvalidWeekdays): dt = get_restricted_datetime(self.dt, weekdays=(7,))
def test_get_naive_datetime(self): dt = datetime(2013,1,1,6) rdt = get_restricted_datetime(dt, '0-24') self.assertEqual(dt, rdt)
def test_get_restricted_weekdays(self, weekdays, result): dt = get_restricted_datetime(self.dt, weekdays=weekdays) self.assertEqual(dt, result)
def test_get_restricted_datetime(self, between, result): dt = get_restricted_datetime(self.dt, between) self.assertEqual(dt, result)