# holidays # -------- # A fast, efficient Python library for generating country, province and state # specific sets of holidays on the fly. It aims to make determining whether a # specific date is a holiday as fast and flexible as possible. # # Authors: Vacanza Team and individual contributors (see CONTRIBUTORS file) # dr-prodigy (c) 2017-2023 # ryanss (c) 2014-2017 # Website: https://github.com/vacanza/holidays # License: MIT (see LICENSE file) import pickle import unittest from datetime import date, datetime from datetime import timedelta as td from holidays.calendars.gregorian import JAN, FEB, OCT, DEC, MON, TUE, SAT, SUN from holidays.constants import HOLIDAY_NAME_DELIMITER, OPTIONAL, PUBLIC, SCHOOL from holidays.countries import UA, US from holidays.groups.christian import ChristianHolidays from holidays.groups.custom import StaticHolidays from holidays.holiday_base import HolidayBase class EntityStubStaticHolidays: special_public_holidays = { 1111: (JAN, 1, "Test holiday"), 2222: (FEB, 2, "Test holiday"), 3333: ((FEB, 2, "Test holiday")), 4444: (), 1991: ( (JAN, 7, JAN, 12), (JAN, 8, JAN, 13, 1991), ), 2024: (FEB, 19, FEB, 24), } substituted_date_format = "%d/%m/%Y" substituted_label = "From %s" class EntityStub(HolidayBase): def _add_observed(self, dt: date, before: bool = True, after: bool = True) -> None: if not self.observed: return None if self._is_saturday(dt) and before: self._add_holiday(f"{self[dt]} (observed)", dt + td(days=-1)) elif self._is_sunday(dt) and after: self._add_holiday(f"{self[dt]} (observed)", dt + td(days=+1)) def _populate(self, year: int) -> None: super()._populate(year) name = "New Year's Day" self._add_observed(self._add_holiday_jan_1(name), before=False) if self.observed and self._is_friday(DEC, 31): self._add_holiday_dec_31(f"{name} (observed)") self._add_observed(self._add_holiday_jun_19("Juneteenth National Independence Day")) self._add_observed(self._add_holiday_jul_4("Independence Day")) self._add_holiday_4th_thu_of_nov("Thanksgiving") self._add_observed(self._add_holiday_dec_25("Christmas Day")) class CountryStub1(EntityStub, StaticHolidays): country = "CS1" subdivisions = ("Subdiv 1", "Subdiv 2", "3") subdivisions_aliases = { "S1": "Subdiv 1", "S2": "Subdiv 2", "S3": "3", "S_1": "Subdiv 1", "S_2": "Subdiv 2", "S_3": "3", } supported_categories = (PUBLIC, SCHOOL) start_year = 1000 end_year = 3400 def __init__(self, *args, **kwargs) -> None: StaticHolidays.__init__(self, cls=EntityStubStaticHolidays) super().__init__(*args, **kwargs) def _populate_subdiv_subdiv_1_public_holidays(self): self._add_holiday_aug_10("Subdiv 1 Custom Holiday") def _populate_subdiv_subdiv_2_public_holidays(self): self._add_holiday_aug_10("Subdiv 2 Custom Holiday") class CountryStub2(EntityStub): country = "CS2" subdivisions = ("Subdiv-3", "Subdiv-4") def _populate(self, year: int) -> None: super()._populate(year) self._add_holiday_mar_1("Custom March 1st Holiday") class CountryStub3(HolidayBase): country = "CS3" def _populate(self, year: int) -> None: super()._populate(year) self._add_holiday_may_1("Custom May 1st Holiday") self._add_holiday_may_2("Custom May 2nd Holiday") class CountryStub4(HolidayBase): country = "CS4" supported_categories = ("CUSTOM_1", "CUSTOM_2") class CountryStub5(HolidayBase, ChristianHolidays): country = "CS5" def __init__(self, *args, **kwargs) -> None: ChristianHolidays.__init__(self) super().__init__(*args, **kwargs) class CountryStub6(EntityStub, StaticHolidays): country = "CS6" def __init__(self, *args, **kwargs) -> None: StaticHolidays.__init__(self, cls=EntityStubStaticHolidays) super().__init__(*args, **kwargs) def _populate_public_holidays(self) -> None: self._add_holiday_may_1("Labor Day") self._add_holiday_may_2("Labor Day Two") class MarketStub1(EntityStub): market = "MS1" class MarketStub2(EntityStub): market = "MS2" class TestArgs(unittest.TestCase): def test_categories(self): self.assertEqual({PUBLIC}, CountryStub1(categories=PUBLIC).categories) self.assertEqual({PUBLIC, SCHOOL}, CountryStub1(categories=(PUBLIC, SCHOOL)).categories) self.assertRaises(ValueError, lambda: CountryStub1(categories=OPTIONAL)) self.assertRaises(ValueError, lambda: CountryStub1(categories="UNSUPPORTED")) self.assertRaises(ValueError, lambda: CountryStub1(categories=("HOME", "UNSUPPORTED"))) self.assertRaises(ValueError, lambda: CountryStub4()) def test_country(self): self.assertEqual(CountryStub1().country, "CS1") def test_expand(self): hb = CountryStub1(years=(2013, 2015), expand=False) self.assertFalse(hb.expand) self.assertSetEqual(hb.years, {2013, 2015}) self.assertNotIn("2014-01-01", hb) self.assertSetEqual(hb.years, {2013, 2015}) def test_observed(self): hb = CountryStub1(observed=False) self.assertFalse(hb.observed) self.assertIn("2000-01-01", hb) self.assertNotIn("1999-12-31", hb) self.assertIn("2012-01-01", hb) self.assertNotIn("2012-01-02", hb) hb.observed = True self.assertIn("2000-01-01", hb) self.assertIn("1999-12-31", hb) self.assertIn("2012-01-01", hb) self.assertIn("2012-01-02", hb) hb.observed = False self.assertIn("2000-01-01", hb) self.assertNotIn("1999-12-31", hb) self.assertIn("2012-01-01", hb) self.assertNotIn("2012-01-02", hb) def test_subdivision(self): self.assertEqual(CountryStub1(subdiv="Subdiv 1").subdiv, "Subdiv 1") self.assertEqual(CountryStub1(subdiv=3).subdiv, "3") def test_subdivisions_aliases(self): subdivisions_aliases = { "subdiv_1": ("S1", "S_1"), "subdiv_2": ("S2", "S_2"), "3": ("S3", "S_3"), } for subdiv, aliases in subdivisions_aliases.items(): for alias in aliases: self.assertEqual(subdiv, CountryStub1(subdiv=alias)._normalized_subdiv) self.assertEqual( CountryStub1.get_subdivision_aliases(), { "Subdiv 1": ["S1", "S_1"], "Subdiv 2": ["S2", "S_2"], "3": ["S3", "S_3"], }, ) def test_years(self): hb = HolidayBase() self.assertSetEqual(hb.years, set()) self.assertNotIn("2014-01-02", hb) self.assertSetEqual(hb.years, {2014}) self.assertNotIn("2013-01-02", hb) self.assertNotIn("2014-01-02", hb) self.assertNotIn("2015-01-02", hb) self.assertSetEqual(hb.years, {2013, 2014, 2015}) self.assertSetEqual( HolidayBase(years=range(2010, 2016)).years, {2010, 2011, 2012, 2013, 2014, 2015} ) self.assertSetEqual(HolidayBase(years=(2013, 2015, 2015)).years, {2013, 2015}) self.assertSetEqual(HolidayBase(years=(2013.0, 2015.0, 2015.0)).years, {2013, 2015}) self.assertSetEqual(HolidayBase(years=2015).years, {2015}) self.assertSetEqual(HolidayBase(years=2015.0).years, {2015}) class TestCategories(unittest.TestCase): class CustomCategoryClass(HolidayBase): country = "CCC" default_category = "CC" subdivisions = ("SD_1", "SD_2") supported_categories = ("CC", "CC_1", "CC_2") def _populate_cc_holidays(self): self._add_holiday_may_15("CC Holiday") def _populate_cc_1_holidays(self): self._add_holiday_may_16("CC1 Holiday") def _populate_cc_2_holidays(self): self._add_holiday_jun_15("CC2 Holiday") def _populate_subdiv_sd_1_cc_holidays(self): self._add_holiday_jul_1("SD_1 CC Holiday") def _populate_subdiv_sd_1_cc_1_holidays(self): self._add_holiday_jul_1("SD_1 CC_1 Holiday") def _populate_subdiv_sd_2_cc_holidays(self): self._add_holiday_aug_1("SD_2 CC Holiday") def test_default_category(self): ccc = TestCategories.CustomCategoryClass(years=2024) self.assertEqual(ccc.categories, {TestCategories.CustomCategoryClass.default_category}) for name in ("CC Holiday",): self.assertTrue(ccc.get_named(name, lookup="exact")) for name in ("CC1 Holiday", "CC2 Holiday", "SD_1 CC_1 Holiday", "SD_2 CC Holiday"): self.assertFalse(ccc.get_named(name, lookup="exact")) # Default category with subdiv. ccc_sd2 = TestCategories.CustomCategoryClass(years=2024, subdiv="SD_2") self.assertEqual(ccc_sd2.categories, {TestCategories.CustomCategoryClass.default_category}) for name in ("CC Holiday", "SD_2 CC Holiday"): self.assertTrue(ccc_sd2.get_named(name, lookup="exact"), name) for name in ("CC1 Holiday", "CC2 Holiday", "SD_1 CC_1 Holiday"): self.assertFalse(ccc_sd2.get_named(name, lookup="exact"), name) def test_no_default_category(self): TestCategories.CustomCategoryClass.default_category = None self.assertRaises(ValueError, lambda: TestCategories.CustomCategoryClass(years=2024)) self.assertRaises( ValueError, lambda: TestCategories.CustomCategoryClass(years=2024, subdiv="SD_1") ) # Explicitly set category. ccc = TestCategories.CustomCategoryClass(years=2024, categories="CC") self.assertEqual(ccc.categories, {"CC"}) self.assertIn("2024-05-15", ccc) def test_non_default_categories(self): for categories in ( {"CC_1": ("2024-05-16",)}, {"CC_2": ("2024-06-15",)}, ): for category in categories: ccc = TestCategories.CustomCategoryClass(years=2024, categories=category) self.assertEqual(ccc.categories, set(categories), category) for dt in categories[category]: self.assertIn(dt, ccc) class TestDeprecationWarnings(unittest.TestCase): def test_prov_deprecation(self): with self.assertWarns(Warning): CountryStub1(prov="Subdiv 1") def test_state_deprecation(self): with self.assertWarns(Warning): CountryStub1(state="Subdiv 1") class TestEqualityInequality(unittest.TestCase): def test_eq(self): hb_1 = CountryStub1() hb_2 = CountryStub2() hb_3 = CountryStub1(subdiv="Subdiv 1") self.assertEqual(hb_1, hb_1) self.assertEqual(hb_2, hb_2) self.assertEqual(hb_3, hb_3) hb_4 = CountryStub1(years=2014) hb_5 = CountryStub1(years=2014) self.assertEqual(hb_4, hb_5) self.assertEqual(hb_5, hb_4) hb_6 = CountryStub2(years=(2014, 2015)) hb_7 = CountryStub2(years=(2014, 2015)) self.assertEqual(hb_6, hb_7) self.assertEqual(hb_7, hb_6) hb_8 = CountryStub3(language="fr") hb_9 = CountryStub3(language="fr") self.assertEqual(hb_8, hb_9) self.assertEqual(hb_9, hb_8) # Use assertFalse instead of assertNotEqual as we want to check "==" explicitly. self.assertFalse(hb_1 == {}) self.assertFalse(hb_2 == {}) self.assertFalse(hb_3 == {}) self.assertFalse(hb_1 == hb_2) self.assertFalse(hb_2 == hb_1) self.assertFalse(hb_1 == hb_3) self.assertFalse(hb_3 == hb_1) self.assertFalse(hb_2 == hb_3) self.assertFalse(hb_3 == hb_2) def test_ne(self): hb_1 = CountryStub1() hb_2 = CountryStub2() hb_3 = CountryStub1(subdiv="Subdiv 1") self.assertNotEqual(hb_1, {}) self.assertNotEqual(hb_2, {}) self.assertNotEqual(hb_3, {}) self.assertNotEqual(hb_1, hb_2) self.assertNotEqual(hb_2, hb_1) self.assertNotEqual(hb_1, hb_3) self.assertNotEqual(hb_3, hb_1) self.assertNotEqual(hb_2, hb_3) self.assertNotEqual(hb_3, hb_2) hb_4 = CountryStub1(years=2014) hb_5 = CountryStub2(years=2015) self.assertNotEqual(hb_1, hb_4) self.assertNotEqual(hb_4, hb_1) self.assertNotEqual(hb_2, hb_5) self.assertNotEqual(hb_5, hb_2) self.assertNotEqual(hb_4, hb_5) self.assertNotEqual(hb_5, hb_4) hb_6 = CountryStub1(years=2014, subdiv="Subdiv 1") hb_7 = CountryStub1(years=2014, subdiv="Subdiv 2") self.assertNotEqual(hb_6, hb_7) self.assertNotEqual(hb_7, hb_6) hb_8 = CountryStub1(years=2014) hb_9 = CountryStub1(years=2014, language="fr") self.assertNotEqual(hb_8, hb_9) self.assertNotEqual(hb_9, hb_8) # Use assertFalse instead of assertEqual in order to check "!=" explicitly. self.assertFalse(hb_1 != hb_1) self.assertFalse(hb_2 != hb_2) self.assertFalse(hb_3 != hb_3) class TestGetList(unittest.TestCase): def test_get_list_multiple_countries(self): hb_country_1 = CountryStub1(years=2021) hb_country_2 = CountryStub1(years=2021) hb_country_1._add_holiday_dec_20("Custom Holiday 1") hb_country_2._add_holiday_dec_20("Custom Holiday 2") self.assertEqual(hb_country_1["2021-12-20"], "Custom Holiday 1") self.assertEqual(hb_country_2["2021-12-20"], "Custom Holiday 2") self.assertListEqual(hb_country_1.get_list("2021-12-20"), ["Custom Holiday 1"]) self.assertListEqual(hb_country_2.get_list("2021-12-20"), ["Custom Holiday 2"]) hb_combined = hb_country_1 + hb_country_2 self.assertEqual(hb_combined["2021-12-20"], "Custom Holiday 1; Custom Holiday 2") self.assertListEqual( hb_combined.get_list("2021-12-20"), ["Custom Holiday 1", "Custom Holiday 2"] ) def test_get_list_multiple_subdivisions(self): hb_subdiv_1 = CountryStub1(subdiv="Subdiv 1", years=2021) hb_subdiv_2 = CountryStub1(subdiv="Subdiv 2", years=2021) self.assertEqual(hb_subdiv_1["2021-08-10"], "Subdiv 1 Custom Holiday") self.assertEqual(hb_subdiv_2["2021-08-10"], "Subdiv 2 Custom Holiday") self.assertListEqual(hb_subdiv_1.get_list("2021-08-10"), ["Subdiv 1 Custom Holiday"]) self.assertListEqual(hb_subdiv_2.get_list("2021-08-10"), ["Subdiv 2 Custom Holiday"]) hb_combined = hb_subdiv_1 + hb_subdiv_2 self.assertEqual( hb_combined["2021-08-10"], "Subdiv 1 Custom Holiday; Subdiv 2 Custom Holiday" ) self.assertListEqual( hb_combined.get_list("2021-08-10"), ["Subdiv 1 Custom Holiday", "Subdiv 2 Custom Holiday"], ) class TestGetNamed(unittest.TestCase): def test_contains(self): hb = CountryStub1(years=2022) for name in ("New", "Year"): self.assertListEqual(hb.get_named(name, lookup="contains"), [date(2022, 1, 1)]) self.assertListEqual( hb.get_named("Independence Day", lookup="contains"), [date(2022, 6, 19), date(2022, 6, 20), date(2022, 7, 4)], ) self.assertListEqual(hb.get_named("independence day", lookup="contains"), []) self.assertListEqual(hb.get_named("Thanksgiving", lookup="contains"), [date(2022, 11, 24)]) self.assertListEqual(hb.get_named("Thanksgivi", lookup="contains"), [date(2022, 11, 24)]) self.assertListEqual(hb.get_named("thanks", lookup="contains"), []) self.assertSetEqual(hb.years, {2022}) hb = CountryStub1(observed=False, years=2022) self.assertListEqual( hb.get_named("Independence Day", lookup="contains"), [date(2022, 6, 19), date(2022, 7, 4)], ) self.assertListEqual(hb.get_named("independence day", lookup="contains"), []) self.assertSetEqual(hb.years, {2022}) def test_exact(self): hb = CountryStub1(years=2022) for name in ("New", "Day"): self.assertListEqual(hb.get_named(name, lookup="exact"), []) self.assertListEqual(hb.get_named("Independence Day", lookup="exact"), [date(2022, 7, 4)]) self.assertListEqual(hb.get_named("Thanksgiving", lookup="exact"), [date(2022, 11, 24)]) self.assertListEqual(hb.get_named("thanksgiving", lookup="exact"), []) self.assertSetEqual(hb.years, {2022}) hb = CountryStub1(observed=False, years=2022) self.assertListEqual(hb.get_named("Independence Day", lookup="exact"), [date(2022, 7, 4)]) self.assertSetEqual(hb.years, {2022}) def test_icontains(self): hb = CountryStub1(years=2022) for name in ("Thanksgiving", "thanksgiving", "Thanksgivi"): self.assertListEqual(hb.get_named(name, lookup="icontains"), [date(2022, 11, 24)]) self.assertListEqual( hb.get_named("Independence Day", lookup="icontains"), [date(2022, 6, 19), date(2022, 6, 20), date(2022, 7, 4)], ) self.assertSetEqual(hb.years, {2022}) hb = CountryStub1(observed=False, years=2022) self.assertListEqual( hb.get_named("Independence Day", lookup="icontains"), [date(2022, 6, 19), date(2022, 7, 4)], ) self.assertSetEqual(hb.years, {2022}) def test_iexact(self): hb = CountryStub1(years=2022) for name in ("new year's day", "New Year's Day"): self.assertListEqual(hb.get_named(name, lookup="iexact"), [date(2022, 1, 1)]) for name in ("New Year Day", "New Year", "year", "NEW Year"): self.assertListEqual(hb.get_named(name, lookup="iexact"), []) self.assertListEqual(hb.get_named("independence day", lookup="iexact"), [date(2022, 7, 4)]) self.assertListEqual(hb.get_named("thanksgiving", lookup="iexact"), [date(2022, 11, 24)]) self.assertListEqual(hb.get_named("Thanksgivin", lookup="iexact"), []) self.assertSetEqual(hb.years, {2022}) hb = CountryStub1(observed=False, years=2022) self.assertListEqual(hb.get_named("independence day", lookup="iexact"), [date(2022, 7, 4)]) self.assertSetEqual(hb.years, {2022}) def test_invalid(self): hb = CountryStub1(years=2022) self.assertRaises( AttributeError, lambda: hb.get_named("Independence Day", lookup="invalid") ) def test_istartswith(self): hb = CountryStub1(years=2022) for name in ("new year's", "New Year's", "New Year's day"): self.assertListEqual(hb.get_named(name, lookup="istartswith"), [date(2022, 1, 1)]) for name in ("New Year Day", "New Year holiday", "New Year's Day Holiday", "year"): self.assertListEqual(hb.get_named(name, lookup="istartswith"), []) self.assertListEqual( hb.get_named("independence day", lookup="istartswith"), [date(2022, 7, 4)] ) self.assertSetEqual(hb.years, {2022}) hb = CountryStub1(observed=False, years=2022) self.assertListEqual( hb.get_named("independence day", lookup="istartswith"), [date(2022, 7, 4)] ) self.assertSetEqual(hb.years, {2022}) def test_startswith(self): hb = CountryStub1(years=2022) for name in ("New Year's", "New Year"): self.assertListEqual(hb.get_named(name, lookup="startswith"), [date(2022, 1, 1)]) for name in ("New Year Day", "New Year Holiday", "New Year's Day Holiday", "year"): self.assertListEqual(hb.get_named(name, lookup="startswith"), []) self.assertListEqual( hb.get_named("Independence Day", lookup="startswith"), [date(2022, 7, 4)] ) self.assertListEqual( hb.get_named("Christmas", lookup="startswith"), [date(2022, 12, 25), date(2022, 12, 26)], ) self.assertSetEqual(hb.years, {2022}) hb = CountryStub1(observed=False, years=2022) self.assertListEqual( hb.get_named("Independence Day", lookup="startswith"), [date(2022, 7, 4)] ) self.assertListEqual(hb.get_named("Christmas", lookup="startswith"), [date(2022, 12, 25)]) self.assertSetEqual(hb.years, {2022}) class TestHelperMethods(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_add_holiday(self): self.hb._populate(2023) self.hb._add_holiday("Test 1", date(2023, JAN, 5)) self.hb._add_holiday("Test 2", (JAN, 6)) self.hb._add_holiday_jan_7("Test 3") self.assertIn("2023-01-05", self.hb) self.assertIn("2023-01-06", self.hb) self.assertIn("2023-01-07", self.hb) for args in ( (date(2020, JAN, 5),), (JAN, 5, "Test 1", True), ("Test", "Test"), ): self.assertRaises(TypeError, lambda: self.hb._add_holiday(*args)) def test_is_leap_year(self): self.hb._populate(1999) self.assertFalse(self.hb._is_leap_year()) self.hb._populate(2000) self.assertTrue(self.hb._is_leap_year()) self.hb._populate(2004) self.assertTrue(self.hb._is_leap_year()) self.hb._populate(2200) self.assertFalse(self.hb._is_leap_year()) def test_is_weekend(self): self.hb._populate(2022) dts = (date(2022, 10, 3), date(2022, 10, 4), (OCT, 3), (OCT, 4)) self.hb.weekend = {MON, TUE} for dt in dts: self.assertTrue(self.hb._is_weekend(dt)) self.hb.weekend = {} for dt in dts: self.assertFalse(self.hb._is_weekend(dt)) self.hb.weekend = {SAT, SUN} for dt in (date(2022, 10, 1), date(2022, 10, 2)): self.assertTrue(self.hb._is_weekend(dt)) for dt in ((OCT, 1), (OCT, 2)): self.assertTrue(self.hb._is_weekend(dt)) self.assertTrue(self.hb._is_weekend(*dt)) for dt in (date(2022, 10, 3), date(2022, 10, 4)): self.assertFalse(self.hb._is_weekend(dt)) for dt in ((OCT, 3), (OCT, 4)): self.assertFalse(self.hb._is_weekend(dt)) self.assertFalse(self.hb._is_weekend(*dt)) class TestHolidaySum(unittest.TestCase): def setUp(self) -> None: self.hb_1 = CountryStub1(years=2014) self.hb_2 = CountryStub2(years=2015) self.hb_3 = CountryStub3() self.hb_combined = self.hb_1 + self.hb_2 + self.hb_3 def assertAdded(self): # noqa: N802 self.assertNotIn("2014-03-01", self.hb_1) self.assertNotIn("2014-05-01", self.hb_1) self.assertNotIn("2014-05-02", self.hb_1) self.assertIn("2014-03-01", self.hb_2) self.assertIn("2014-05-01", self.hb_3) self.assertIn("2014-05-02", self.hb_3) self.assertIn("2014-07-04", self.hb_1) self.assertIn("2014-07-04", self.hb_1 + self.hb_2) self.assertIn("2014-07-04", self.hb_2 + self.hb_1) self.assertIn("2015-07-04", self.hb_1 + self.hb_2) self.assertIn("2015-07-04", self.hb_2 + self.hb_1) self.assertIn("2000-03-01", self.hb_combined) self.assertIn("2000-05-01", self.hb_combined) self.assertIn("2000-05-02", self.hb_combined) self.assertIn("2000-07-04", self.hb_combined) self.assertIn("2000-07-04", self.hb_combined) def test_add_country(self): self.assertAdded() self.hb_combined = CountryStub1(years=2014) self.hb_combined += CountryStub2(years=2015) self.hb_combined += CountryStub3() self.assertAdded() self.hb_combined = ( self.hb_1 + (self.hb_2 + self.hb_3) + self.hb_1 + (self.hb_3 + self.hb_2 + CountryStub2(subdiv="Subdiv-4")) ) self.assertAdded() self.hb_combined = CountryStub1(years=2014, subdiv="Subdiv 1") self.hb_combined += CountryStub1(years=2014, subdiv="Subdiv 2") self.assertEqual( self.hb_combined["2014-08-10"], "Subdiv 1 Custom Holiday; Subdiv 2 Custom Holiday" ) self.assertRaises(TypeError, lambda: self.hb_1 + {}) def test_add_financial(self): hb_1 = MarketStub1(years=2013) hb_2 = MarketStub2(years=(2014, 2015)) hb_combined = hb_1 + hb_2 self.assertIn("2013-01-01", hb_1) self.assertIn("2014-01-01", hb_2) self.assertIn("2015-01-01", hb_2) self.assertIn("2013-01-01", hb_combined) self.assertIn("2014-01-01", hb_combined) self.assertIn("2015-01-01", hb_combined) self.assertListEqual(hb_combined.market, ["MS1", "MS2"]) def test_args(self): self.hb_1 = CountryStub1(years=2014, subdiv="Subdiv 1") self.hb_2 = CountryStub2(years=2015, subdiv="Subdiv-3") self.assertListEqual(self.hb_combined.country, ["CS1", "CS2", "CS3"]) self.assertListEqual((self.hb_1 + self.hb_2).subdiv, ["Subdiv 1", "Subdiv-3"]) self.assertListEqual((self.hb_2 + self.hb_1).subdiv, ["Subdiv-3", "Subdiv 1"]) self.assertEqual((self.hb_1 + self.hb_3).subdiv, "Subdiv 1") self.assertTrue((self.hb_2 + self.hb_3).expand) self.assertSetEqual((self.hb_2 + self.hb_3).years, {2014, 2015}) self.assertSetEqual((self.hb_3 + self.hb_2).years, {2014, 2015}) self.hb_combined = sum(CountryStub1(subdiv=subdiv) for subdiv in CountryStub1.subdivisions) self.assertEqual(self.hb_combined.country, CountryStub1.country) self.assertEqual(self.hb_combined.subdiv, list(CountryStub1.subdivisions)) class TestInheritance(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_custom_holidays(self): class CustomHolidays(CountryStub1): def _populate(self, year): super()._populate(year) self._add_holiday_jul_13("Ninja Turtle's Day") self._add_holiday_dec_31("New Year's Eve") del self[date(year, JAN, 1)] hb = CustomHolidays(years=(2014, 2020)) self.assertIn("2014-01-01", self.hb) self.assertNotIn("2014-01-01", hb) self.assertIn("2014-12-31", hb) self.assertNotIn("2014-12-31", self.hb) self.assertIn("2020-01-01", self.hb) self.assertNotIn("2020-01-01", hb) self.assertIn("2020-12-31", hb) self.assertNotIn("2020-12-31", self.hb) self.assertNotIn("2014-07-13", self.hb) self.assertIn("2014-07-13", hb) self.assertNotIn("2020-07-13", self.hb) self.assertIn("2020-07-13", hb) class TestKeyTransforms(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_date(self): dt = date(2014, 1, 1) self.assertIn(dt, self.hb) self.assertEqual(self.hb[dt], "New Year's Day") dt = date(2014, 1, 3) self.hb[dt] = "Fake Holiday" self.assertIn(dt, self.hb) self.assertEqual(self.hb.pop(dt), "Fake Holiday") self.assertNotIn(dt, self.hb) def test_date_subclass(self): class CustomDateType(date): pass self.assertTrue(issubclass(CustomDateType, date)) self.assertIn(CustomDateType(2014, 1, 1), self.hb) self.assertNotIn(CustomDateType(2014, 1, 3), self.hb) def test_datetime(self): self.assertIn(datetime(2014, 1, 1, 13, 45), self.hb) self.assertEqual(self.hb[datetime(2014, 1, 1, 13, 45)], "New Year's Day") self.hb[datetime(2014, 1, 3, 1, 1)] = "Fake Holiday" self.assertIn(datetime(2014, 1, 3, 2, 2), self.hb) self.assertEqual(self.hb.pop(datetime(2014, 1, 3, 4, 4)), "Fake Holiday") self.assertNotIn(datetime(2014, 1, 3, 2, 2), self.hb) def test_exception(self): self.assertRaises((TypeError, ValueError), lambda: "abc" in self.hb) self.assertRaises((TypeError, ValueError), lambda: self.hb.get("abc123")) self.assertRaises(TypeError, lambda: self.hb.get({"123"})) self.assertRaises((TypeError, ValueError), self.hb.__setitem__, "abc", "Test") self.assertRaises((TypeError, ValueError), lambda: {} in self.hb) def test_string(self): self.assertIn("2014-01-01", self.hb) self.assertEqual(self.hb["2014-01-01"], "New Year's Day") self.assertIn("01/01/2014", self.hb) self.assertEqual(self.hb["01/01/2014"], "New Year's Day") self.hb["01/03/2014"] = "Fake Holiday" self.assertIn("01/03/2014", self.hb) self.assertNotIn("03/01/2014", self.hb) self.assertIn("2014-01-03", self.hb) self.assertNotIn("2014-03-01", self.hb) self.assertEqual(self.hb.pop("01/03/2014"), "Fake Holiday") def test_timestamp(self): self.assertIn(1388552400, self.hb) self.assertEqual(self.hb[1388552400], "New Year's Day") self.assertIn(1388552400.01, self.hb) self.assertEqual(self.hb[1388552400.01], "New Year's Day") self.hb[1388725200] = "Fake Holiday" self.assertIn(1388725201, self.hb) self.assertEqual(self.hb.pop(1388725202), "Fake Holiday") self.assertNotIn(1388725201, self.hb) class TestPop(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_exception(self): self.assertRaises(KeyError, lambda: self.hb.pop("2014-01-02")) self.hb.pop("2014-01-01") self.assertRaises(KeyError, lambda: self.hb.pop("2014-01-01")) def test_success(self): self.assertFalse(self.hb.pop("2014-01-02", False)) self.assertTrue(self.hb.pop("2014-01-02", True)) self.assertIn("2014-01-01", self.hb) self.assertEqual(self.hb.pop("2014-01-01"), "New Year's Day") self.assertNotIn("2014-01-01", self.hb) class TestPopNamed(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_exception(self): self.assertRaises(KeyError, lambda: self.hb.pop_named("New Year's Dayz")) self.assertIn("2014-01-01", self.hb) self.hb.pop_named("New Year's Day") self.assertRaises(KeyError, lambda: self.hb.pop_named("New Year's Day")) def test_single(self): self.assertIn("2014-01-01", self.hb) for dt in self.hb.pop_named("New Year's Day"): self.assertNotIn(dt, self.hb) def test_multiple(self): dt = date(2022, 2, 22) holiday_name_1 = "Holiday Name 1" holiday_name_2 = "Holiday Name 2" holiday_name_3 = "Holiday Name 3" combined_name = HOLIDAY_NAME_DELIMITER.join( (holiday_name_1, holiday_name_2, holiday_name_3) ) self.hb[dt] = holiday_name_1 self.hb[dt] = holiday_name_2 self.hb[dt] = holiday_name_3 self.assertEqual(self.hb[dt], combined_name) # Pop the entire date by multiple holidays exact name. self.hb.pop_named(combined_name) self.assertNotIn(dt, self.hb) # Pop only one holiday by a single name. self.hb[dt] = holiday_name_1 self.hb[dt] = holiday_name_2 self.hb[dt] = holiday_name_3 # 3 holidays names. self.assertEqual(self.hb[dt], combined_name) self.hb.pop_named(holiday_name_1) # 2 holiday names. self.assertEqual( self.hb[dt], HOLIDAY_NAME_DELIMITER.join((holiday_name_2, holiday_name_3)) ) self.hb.pop_named(holiday_name_3) # 1 holiday name. self.assertEqual(self.hb[dt], holiday_name_2) self.hb.pop_named(holiday_name_2) # 0 holiday names. self.assertNotIn(dt, self.hb) def test_partial(self): self.assertIn("2014-01-01", self.hb) for dt in self.hb.pop_named("N"): self.assertNotIn(dt, self.hb) self.assertRaises(KeyError, lambda: self.hb.pop_named("New Year")) def test_contains(self): self.assertIn("2022-01-01", self.hb) removed_dates = self.hb.pop_named("Day", lookup="contains") self.assertEqual(len(removed_dates), 6) self.assertNotIn("2022-01-01", self.hb) self.assertNotIn("2022-06-19", self.hb) self.assertNotIn("2022-06-20", self.hb) self.assertNotIn("2022-07-04", self.hb) self.assertNotIn("2022-12-25", self.hb) self.assertNotIn("2022-12-26", self.hb) def test_icontains(self): self.assertIn("2022-01-01", self.hb) removed_dates = self.hb.pop_named("day", lookup="icontains") self.assertEqual(len(removed_dates), 6) self.assertNotIn("2022-01-01", self.hb) self.assertNotIn("2022-06-19", self.hb) self.assertNotIn("2022-06-20", self.hb) self.assertNotIn("2022-07-04", self.hb) self.assertNotIn("2022-12-25", self.hb) self.assertNotIn("2022-12-26", self.hb) def test_exact(self): self.assertIn("2022-01-01", self.hb) removed_dates = self.hb.pop_named("Independence Day", lookup="exact") self.assertEqual(len(removed_dates), 1) self.assertNotIn("2022-07-04", self.hb) self.assertIn("2022-06-19", self.hb) self.assertIn("2022-06-20", self.hb) hb = CountryStub1(years=2025) hb["2024-02-02"] = "Big Groundhog Day; Groundhog Day" removed_dates = hb.pop_named("Groundhog Day", lookup="exact") self.assertEqual(len(removed_dates), 1) self.assertEqual(hb["2024-02-02"], "Big Groundhog Day") def test_iexact(self): self.assertIn("2022-01-01", self.hb) removed_dates = self.hb.pop_named("independence day", lookup="iexact") self.assertEqual(len(removed_dates), 1) self.assertNotIn("2022-07-04", self.hb) self.assertIn("2022-06-19", self.hb) self.assertIn("2022-06-20", self.hb) hb = CountryStub1(years=2025) hb["2024-02-02"] = "Big Groundhog Day; Groundhog Day" removed_dates = hb.pop_named("Groundhog day", lookup="iexact") self.assertEqual(len(removed_dates), 1) self.assertEqual(hb["2024-02-02"], "Big Groundhog Day") def test_startswith(self): self.assertIn("2022-01-01", self.hb) removed_dates = self.hb.pop_named("Independence", lookup="startswith") self.assertEqual(len(removed_dates), 1) self.assertNotIn("2022-07-04", self.hb) self.assertIn("2022-06-19", self.hb) self.assertIn("2022-06-20", self.hb) removed_dates = self.hb.pop_named("Christmas", lookup="startswith") self.assertEqual(len(removed_dates), 2) self.assertNotIn("2022-12-25", self.hb) self.assertNotIn("2022-12-26", self.hb) def test_istartswith(self): self.assertIn("2022-01-01", self.hb) removed_dates = self.hb.pop_named("independence", lookup="istartswith") self.assertEqual(len(removed_dates), 1) self.assertNotIn("2022-07-04", self.hb) self.assertIn("2022-06-19", self.hb) self.assertIn("2022-06-20", self.hb) removed_dates = self.hb.pop_named("christmas", lookup="istartswith") self.assertEqual(len(removed_dates), 2) self.assertNotIn("2022-12-25", self.hb) self.assertNotIn("2022-12-26", self.hb) class TestRepr(unittest.TestCase): def test_base(self): self.assertEqual(repr(HolidayBase()), "holidays.HolidayBase()") def test_country(self): self.assertEqual(repr(CountryStub1()), "holidays.country_holidays('CS1')") self.assertEqual( repr(CountryStub1(subdiv="Subdiv 1")), "holidays.country_holidays('CS1', subdiv='Subdiv 1')", ) def test_market(self): self.assertEqual(repr(MarketStub1()), "holidays.financial_holidays('MS1')") class TestSerialization(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_pickle(self): dt = "2020-01-01" self.assertIn(dt, self.hb) loaded_holidays = pickle.loads(pickle.dumps(self.hb)) self.assertEqual(loaded_holidays, self.hb) self.assertIn(dt, self.hb) def test_pickle_localized_entity(self): for lang in ("uk", "en_US", None): ua = UA(language=lang) dt = "2021-01-01" self.assertIn(dt, self.hb) loaded_ua = pickle.loads(pickle.dumps(ua)) self.assertEqual(loaded_ua, ua) self.assertIn(dt, loaded_ua) class TestSpecialHolidays(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_populate_special_holidays(self): self.assertSetEqual(self.hb.years, set()) self.hb._populate(1111) self.assertSetEqual(self.hb.years, {1111}) self.assertIn("1111-01-01", self.hb) self.assertIn("2222-02-02", self.hb) self.assertIn("3333-02-02", self.hb) self.assertSetEqual(self.hb.years, {1111, 2222, 3333}) class TestStandardMethods(unittest.TestCase): def setUp(self): self.hb = CountryStub1() def test_append(self): self.hb.append([date(2015, 4, 1), "2015-04-03"]) self.hb.append(date(2015, 4, 6)) self.hb.append("2015-04-07") for dt in ( "2015-01-01", "2015-04-01", "2015-04-03", "2015-04-06", "2015-04-07", "2015-12-25", ): self.assertIn(dt, self.hb) for dt in ( "2015-04-04", "2015-04-05", "2015-04-02", ): self.assertNotIn(dt, self.hb) def test_bool(self): self.assertFalse(self.hb) self.assertEqual(bool(self.hb), False) self.assertEqual(len(self.hb), 0) self.assertIn("2014-01-01", self.hb) self.assertTrue(self.hb) self.assertEqual(bool(self.hb), True) self.assertNotEqual(len(self.hb), 0) def test_contains(self): self.assertIn("2014-01-01", self.hb) self.assertNotIn("2014-01-03", self.hb) def test_copy(self): hb = CountryStub1() self.assertEqual(hb, hb.copy()) hb_fr = CountryStub1(language="fr") hb_xx = CountryStub1(language="xx") self.assertNotEqual(hb, hb_fr) self.assertNotEqual(hb.copy(), hb_fr.copy()) self.assertNotEqual(hb, hb_xx) self.assertNotEqual(hb.copy(), hb_xx.copy()) def test_get(self): self.assertEqual(self.hb.get("2014-01-01"), "New Year's Day") self.assertIsNone(self.hb.get("2014-01-03")) self.assertFalse(self.hb.get("2014-01-03", False)) self.assertTrue(self.hb.get("2014-01-03", True)) def test_getattr(self): self.hb._populate(2023) name = "Test" self.assertEqual(self.hb._add_holiday_3rd_mon_of_jun(name), date(2023, 6, 19)) self.assertEqual(self.hb._add_holiday_4th_fri_of_aug(name), date(2023, 8, 25)) self.assertEqual(self.hb._add_holiday_1st_wed_of_aug(name), date(2023, 8, 2)) self.assertEqual(self.hb._add_holiday_1st_mon_before_may_24(name), date(2023, 5, 22)) self.assertEqual(self.hb._add_holiday_1st_sun_from_aug_31(name), date(2023, 9, 3)) h_chr = CountryStub5() h_chr._populate(2023) self.assertEqual(h_chr._add_holiday_8_days_past_easter(name), date(2023, 4, 17)) self.assertEqual(h_chr._add_holiday_5_days_prior_easter(name), date(2023, 4, 4)) self.assertRaises(ValueError, lambda: self.hb._add_holiday_5th_fri_of_aug(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_4th_nam_of_aug(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_nam_12(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_1st_fri_before_nam_29(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_1st_fri_random_jan_29(name)) self.assertRaises(AttributeError, lambda: h_chr._add_holiday_18_days_past_eastr(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_fe_15(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_2_may_15(name)) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_3th_su_of_aug(name)) self.assertRaises( AttributeError, lambda: self.hb._add_holiday_2_days_past_1st_sat_of_ap(name) ) self.assertRaises(AttributeError, lambda: self.hb._add_holiday_1st_sat_from_fe_10(name)) def test_getitem(self): self.assertEqual(self.hb["2014-01-01"], "New Year's Day") self.assertEqual(self.hb.get("2014-01-01"), "New Year's Day") self.assertIsNone(self.hb.get("2014-01-03")) self.assertRaises(KeyError, lambda: self.hb["2014-01-03"]) self.assertListEqual(self.hb["2013-12-31":"2014-01-02"], [date(2014, 1, 1)]) self.assertListEqual( self.hb["2013-12-24":"2014-01-02"], [date(2013, 12, 25), date(2014, 1, 1)] ) self.assertListEqual(self.hb["2013-12-25":"2014-01-02":3], [date(2013, 12, 25)]) self.assertListEqual( self.hb["2013-12-25":"2014-01-02":7], [date(2013, 12, 25), date(2014, 1, 1)] ) self.assertListEqual(self.hb["2014-01-02":"2013-12-30"], [date(2014, 1, 1)]) self.assertListEqual(self.hb["2014-01-02":"2013-12-25"], [date(2014, 1, 1)]) self.assertListEqual( self.hb["2014-01-02":"2013-12-24"], [date(2014, 1, 1), date(2013, 12, 25)] ) self.assertListEqual(self.hb["2014-01-01":"2013-12-24":3], [date(2014, 1, 1)]) self.assertListEqual( self.hb["2014-01-01":"2013-12-24":7], [date(2014, 1, 1), date(2013, 12, 25)] ) self.assertListEqual(self.hb["2013-12-31":"2014-01-02":-3], []) self.assertListEqual(self.hb["2014-01-01" : "2013-12-24" : td(days=3)], [date(2014, 1, 1)]) self.assertListEqual( self.hb["2014-01-01" : "2013-12-24" : td(days=7)], [date(2014, 1, 1), date(2013, 12, 25)], ) self.assertListEqual(self.hb["2013-12-31" : "2014-01-02" : td(days=3)], []) self.assertRaises(ValueError, lambda: self.hb["2014-01-01":]) self.assertRaises(ValueError, lambda: self.hb[:"2014-01-01"]) self.assertRaises(TypeError, lambda: self.hb["2014-01-01":"2014-01-02":""]) self.assertRaises(ValueError, lambda: self.hb["2014-01-01":"2014-01-02":0]) def test_radd(self): self.assertRaises(TypeError, lambda: 1 + CountryStub1()) def test_setitem(self): self.assertEqual(len(self.hb), 0) self.hb["2014-01-03"] = "Custom Holiday" self.assertGreater(len(self.hb), 0) self.assertIn("2014-01-03", self.hb) self.assertEqual(self.hb["2014-01-03"], "Custom Holiday") self.hb["2014-01-04"] = "Custom Holiday" self.hb["2014-01-04"] = "Custom Holiday; Another Custom Holiday" self.assertEqual(self.hb["2014-01-04"], "Another Custom Holiday; Custom Holiday") def test_update(self): self.hb.update( { "2015-01-10": "Custom Holiday", "2015-01-11": "Custom Holiday", } ) self.assertIn("2015-01-10", self.hb) self.assertIn("2015-01-11", self.hb) self.assertEqual(self.hb["2015-01-10"], "Custom Holiday") self.assertEqual(self.hb["2015-01-11"], "Custom Holiday") self.hb.update( { date(2015, 1, 10): "New Holiday", date(2015, 1, 11): "New Holiday", } ) self.assertIn(date(2015, 1, 10), self.hb) self.assertIn(date(2015, 1, 11), self.hb) self.assertEqual(self.hb["2015-01-10"], "Custom Holiday; New Holiday") self.assertEqual(self.hb["2015-01-11"], "Custom Holiday; New Holiday") class TestStr(unittest.TestCase): def test_country(self): hb = CountryStub1() self.assertEqual( str(hb), "{'country': CS1, 'expand': True, 'language': None, " "'market': None, 'observed': True, 'subdiv': None, " "'years': set()}", ) hb._populate(2013) self.assertEqual( str(hb), '{datetime.date(2013, 1, 1): "New Year\'s Day", ' "datetime.date(2013, 6, 19): 'Juneteenth National Independence Day', " "datetime.date(2013, 7, 4): 'Independence Day', " "datetime.date(2013, 11, 28): 'Thanksgiving', " "datetime.date(2013, 12, 25): 'Christmas Day'}", ) def test_market(self): self.assertEqual( str(MarketStub1()), "{'country': None, 'expand': True, 'language': None, " "'market': MS1, 'observed': True, 'subdiv': None, " "'years': set()}", ) class TestSubstitutedHolidays(unittest.TestCase): class CountryStub(CountryStub2, StaticHolidays): country = "HB" def __init__(self, cls, *args, **kwargs) -> None: StaticHolidays.__init__(self, cls) super().__init__(*args, **kwargs) def test_populate_substituted_holidays(self): hb = CountryStub1() self.assertTrue(hb.has_special_holidays) self.assertTrue(hb.has_substituted_holidays) hb._populate(1991) self.assertIn("1991-01-07", hb) self.assertIn("1991-01-08", hb) self.assertIn("12/01/1991", hb["1991-01-07"]) self.assertIn("13/01/1991", hb["1991-01-08"]) def test_no_substituted_date_format(self): class SubstitutedHolidays: special_public_holidays = { 1991: ( (JAN, 12, JAN, 7), (1991, JAN, 13, JAN, 8), ), } substituted_label = "From %s" self.assertRaises(ValueError, lambda: self.CountryStub(SubstitutedHolidays)) def test_no_substituted_holidays(self): class EmptySubstitutedHolidays: special_public_holidays = {} substituted_date_format = "%d/%m/%Y" substituted_label = "From %s" class NoSubstitutedHolidays: substituted_date_format = "%d/%m/%Y" substituted_label = "From %s" for cls in (EmptySubstitutedHolidays, NoSubstitutedHolidays): hb = self.CountryStub(cls=cls) self.assertFalse(hb.has_special_holidays) self.assertTrue(hb.has_substituted_holidays) hb._populate(1991) self.assertNotIn("1991-01-07", hb) self.assertNotIn("1991-01-08", hb) def test_no_substituted_label(self): class SubstitutedHolidays: special_public_holidays = { 1991: ( (JAN, 7, JAN, 12), (JAN, 8, JAN, 13, 1991), ), } substituted_date_format = "%d/%m/%Y" self.assertRaises(ValueError, lambda: self.CountryStub(SubstitutedHolidays)) class TestWorkdays(unittest.TestCase): def setUp(self): self.hb = CountryStub6(years=2024) def test_is_working_day(self): self.assertTrue(self.hb.is_working_day("2024-02-12")) self.assertFalse(self.hb.is_working_day("2024-02-17")) self.assertFalse(self.hb.is_working_day("2024-02-19")) self.assertTrue(self.hb.is_working_day("2024-02-24")) self.assertTrue(self.hb.is_working_day("2024-04-30")) self.assertFalse(self.hb.is_working_day("2024-05-01")) self.assertFalse(self.hb.is_working_day("2024-05-02")) self.assertTrue(self.hb.is_working_day("2024-05-03")) def test_get_nth_working_day(self): self.assertEqual(self.hb.get_nth_working_day("2024-01-04", 0), date(2024, 1, 4)) self.assertEqual(self.hb.get_nth_working_day("2024-01-04", +1), date(2024, 1, 5)) self.assertEqual(self.hb.get_nth_working_day("2024-01-04", +3), date(2024, 1, 9)) self.assertEqual(self.hb.get_nth_working_day("2024-01-06", +1), date(2024, 1, 8)) self.assertEqual(self.hb.get_nth_working_day("2024-01-26", -10), date(2024, 1, 12)) self.assertEqual(self.hb.get_nth_working_day("2024-01-21", -1), date(2024, 1, 19)) self.assertEqual(self.hb.get_nth_working_day("2024-02-15", +4), date(2024, 2, 22)) self.assertEqual(self.hb.get_nth_working_day("2024-02-15", +5), date(2024, 2, 23)) self.assertEqual(self.hb.get_nth_working_day("2024-02-15", +6), date(2024, 2, 24)) self.assertEqual(self.hb.get_nth_working_day("2024-02-15", +7), date(2024, 2, 26)) self.assertEqual(self.hb.get_nth_working_day("2024-02-26", -7), date(2024, 2, 15)) self.assertEqual(self.hb.get_nth_working_day("2024-02-25", -7), date(2024, 2, 15)) self.assertEqual(self.hb.get_nth_working_day("2024-04-29", +1), date(2024, 4, 30)) self.assertEqual(self.hb.get_nth_working_day("2024-04-29", +2), date(2024, 5, 3)) self.assertEqual(self.hb.get_nth_working_day("2024-04-29", +3), date(2024, 5, 6)) self.assertEqual(self.hb.get_nth_working_day("2024-04-29", +4), date(2024, 5, 7)) self.assertEqual(self.hb.get_nth_working_day("2024-05-10", -10), date(2024, 4, 24)) self.assertEqual(self.hb.get_nth_working_day("2024-05-10", -7), date(2024, 4, 29)) self.assertEqual(self.hb.get_nth_working_day("2024-05-10", -5), date(2024, 5, 3)) self.assertEqual(self.hb.get_nth_working_day("2024-07-27", 0), date(2024, 7, 29)) self.assertEqual(self.hb.get_nth_working_day("2024-07-27", 1), date(2024, 7, 29)) self.assertEqual(self.hb.get_nth_working_day("2024-07-29", 0), date(2024, 7, 29)) def test_get_working_days_count(self): self.assertEqual(self.hb.get_working_days_count("2024-01-03", "2024-01-23"), 15) self.assertEqual(self.hb.get_working_days_count("2024-01-23", "2024-01-03"), 15) self.assertEqual(self.hb.get_working_days_count("2024-01-06", "2024-01-07"), 0) self.assertEqual(self.hb.get_working_days_count("2024-01-16", "2024-01-16"), 1) self.assertEqual(self.hb.get_working_days_count("2024-02-08", "2024-02-15"), 6) self.assertEqual(self.hb.get_working_days_count("2024-02-15", "2024-02-22"), 5) self.assertEqual(self.hb.get_working_days_count("2024-02-22", "2024-02-29"), 7) self.assertEqual(self.hb.get_working_days_count("2024-04-29", "2024-05-03"), 3) self.assertEqual(self.hb.get_working_days_count("2024-04-29", "2024-05-04"), 3) self.assertEqual(self.hb.get_working_days_count("2024-04-29", "2024-05-05"), 3) self.assertEqual(self.hb.get_working_days_count("2024-04-29", "2024-05-06"), 4) class TestClosestHoliday(unittest.TestCase): def setUp(self): self.current_year = datetime.now().year self.next_year = self.current_year + 1 self.previous_year = self.current_year - 1 self.hb = CountryStub3(years=self.current_year) self.next_labor_day_year = ( self.current_year if datetime.now().date() <= self.hb.get_named("Custom May 1st Holiday")[0] else self.next_year ) self.previous_labor_day_year = ( self.current_year if datetime.now().date() >= self.hb.get_named("Custom May 2nd Holiday")[0] else self.previous_year ) def test_get_closest_holiday_forward(self): self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-01-01"), (date(self.current_year, 5, 1), "Custom May 1st Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-04-30"), (date(self.current_year, 5, 1), "Custom May 1st Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-05-01"), (date(self.current_year, 5, 2), "Custom May 2nd Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-05-02"), (date(self.next_year, 5, 1), "Custom May 1st Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.next_year}-01-01"), (date(self.next_year, 5, 1), "Custom May 1st Holiday"), ) self.assertIn( self.hb.get_closest_holiday(), [ (date(self.next_labor_day_year, 5, 1), "Custom May 1st Holiday"), (date(self.next_labor_day_year, 5, 2), "Custom May 2nd Holiday"), ], ) def test_get_closest_holiday_backward(self): self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-12-31", direction="backward"), (date(self.current_year, 5, 2), "Custom May 2nd Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-05-02", direction="backward"), (date(self.current_year, 5, 1), "Custom May 1st Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.current_year}-04-30", direction="backward"), (date(self.previous_year, 5, 2), "Custom May 2nd Holiday"), ) self.assertEqual( self.hb.get_closest_holiday(f"{self.previous_year}-12-31", direction="backward"), (date(self.previous_year, 5, 2), "Custom May 2nd Holiday"), ) self.assertIn( self.hb.get_closest_holiday(direction="backward"), [ (date(self.previous_labor_day_year, 5, 2), "Custom May 2nd Holiday"), (date(self.current_year, 5, 1), "Custom May 1st Holiday"), ], ) def test_get_closest_holiday_corner_cases(self): us = US() # check for date before start of calendar self.assertIsNone(us.get_closest_holiday("1777-01-01", direction="backward")) # check for date after end of calendar self.assertIsNone(us.get_closest_holiday("2100-12-31")) def test_get_closest_holiday_after_empty_year(self): ua = UA(years=2025) # check for date if a year has no holidays ua._add_holiday_jan_1("Custom holiday") self.assertEqual( ua.get_closest_holiday("2022-03-08"), (date(2025, 1, 1), "Custom holiday") ) def test_get_closest_holiday_unsorted_calendars(self): us_calendar = US(years=2024) self.assertEqual( us_calendar.get_closest_holiday(date(2024, 2, 1)), (date(2024, 2, 19), "Washington's Birthday"), ) # check for date before start of calendar self.assertEqual( us_calendar.get_closest_holiday(date(2024, 2, 1), direction="backward"), (date(2024, 1, 15), "Martin Luther King Jr. Day"), ) def test_get_closest_holiday_invalid_direction(self): self.assertRaises( AttributeError, lambda: HolidayBase().get_closest_holiday(direction="invalid") )