def test_excluding(self): dr = DateRange(self.date1, self.date2, self.delta) excludes = [date(2015, 3, 8), date(2015, 3, 22)] dates = [d for d in dr.excluding(excludes)] self.assertEqual(dates, [ date(2015, 3, 1), date(2015, 3, 15), date(2015, 3, 29) ])
def test_enumerate(self): dr = DateRange(self.date1, self.date2, self.delta) dates = [d for d in dr.enumerate()] self.assertEqual(dates, [ (0, date(2015, 3, 1)), (1, date(2015, 3, 8)), (2, date(2015, 3, 15)), (3, date(2015, 3, 22)), (4, date(2015, 3, 29)) ])
def test_reverse(self): dr = DateRange(self.date1, self.date2, self.delta) dr.reverse() dates = [d for d in dr] self.assertEqual(dates, [ date(2015, 3, 31), date(2015, 3, 24), date(2015, 3, 17), date(2015, 3, 10), date(2015, 3, 3) ])
def test_count(self): dr = DateRange(self.date1, self.date2, self.delta) self.assertEqual(dr.count(date(2015, 3, 15)), 1)
def test_index(self): dr = DateRange(self.date1, self.date2, self.delta) self.assertEqual(dr.index(date(2015, 3, 15)), 2)