class WeeklyEventTest(unittest.TestCase):
def setUp(self):
self.june = datetime.date(2015, month=6, day=1)
self.event = WeeklyEvent(title='foo', start_date=self.june,
day_of_the_week=DayOfTheWeek.monday, skip_weeks=0)
def test_is_on_date_no_skip(self):
self.assertTrue(self.event.is_on_date(self.june.replace(day=1)))
self.assertTrue(self.event.is_on_date(self.june.replace(day=8)))
self.assertTrue(self.event.is_on_date(self.june.replace(day=15)))
def test_is_on_date_with_skip_1(self):
self.event.skip_weeks = 1
self.assertTrue(self.event.is_on_date(self.june.replace(day=1)))
self.assertFalse(self.event.is_on_date(self.june.replace(day=8)))
self.assertTrue(self.event.is_on_date(self.june.replace(day=15)))
def test_is_on_date_with_skip_3(self):
self.event.skip_weeks = 3
self.assertTrue(self.event.is_on_date(self.june.replace(day=1)))
self.assertFalse(self.event.is_on_date(self.june.replace(day=8)))
self.assertFalse(self.event.is_on_date(self.june.replace(day=15)))
self.assertFalse(self.event.is_on_date(self.june.replace(day=22)))
self.assertTrue(self.event.is_on_date(self.june.replace(day=29)))
def setUp(self):
self.june = datetime.date(2015, month=6, day=1)
self.event = WeeklyEvent(title='foo', start_date=self.june,
day_of_the_week=DayOfTheWeek.monday, skip_weeks=0)