def test_holiday_list_acces_and_modification(self):
ukcal = UnitedKingdom()
holidays = list(
holiday_list(ukcal, Date(1, Jan, 2011), Date(31, 12, 2011)))
self.assertEquals(8, len(holidays))
new_holiday_date = Date(23, August, 2011)
ukcal.add_holiday(new_holiday_date)
holidays = list(
holiday_list(ukcal, Date(1, Jan, 2011), Date(31, 12, 2011)))
self.assertEquals(9, len(holidays))
ukcal.remove_holiday(new_holiday_date)
holidays = list(
holiday_list(ukcal, Date(1, Jan, 2011), Date(31, 12, 2011)))
self.assertEquals(8, len(holidays))
def test_iteration_on_date_list(self):
date_iterator = holiday_list(
TARGET(), Date(1, Jan, 2000), Date(1, Jan, 2001)
)
holidays = [
Date(21, Apr, 2000), Date(24, Apr, 2000),
Date(1, May, 2000), Date(25, Dec, 2000),
Date(26, Dec, 2000), Date(1, Jan, 2001)
]
for date in date_iterator:
self.assertIn(date, holidays)
def test_holiday_list_acces_and_modification(self):
ukcal = UnitedKingdom()
holidays = list(
holiday_list(ukcal, Date(1, Jan, 2011), Date(31, 12,2011) )
)
self.assertEquals(8, len(holidays))
new_holiday_date = Date(23, August, 2011)
ukcal.add_holiday(new_holiday_date)
holidays = list(
holiday_list(ukcal, Date(1, Jan, 2011), Date(31, 12,2011) )
)
self.assertEquals(9, len(holidays))
ukcal.remove_holiday(new_holiday_date)
holidays = list(
holiday_list(ukcal, Date(1, Jan, 2011), Date(31, 12,2011) )
)
self.assertEquals(8, len(holidays))
def test_iteration_on_date_list(self):
date_iterator = holiday_list(TARGET(), Date(1, Jan, 2000),
Date(1, Jan, 2001))
holidays = [
Date(21, Apr, 2000),
Date(24, Apr, 2000),
Date(1, May, 2000),
Date(25, Dec, 2000),
Date(26, Dec, 2000),
Date(1, Jan, 2001)
]
for date in date_iterator:
self.assertIn(date, holidays)
