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)
