added exclude_dates argument and test for holiday class

Author: sharkipelago

pandas/tests/tseries/holiday/test_holiday.py

@@ -353,3 +353,57 @@ def test_holidays_with_timezone_specified_but_no_occurences():
     expected_results.index = expected_results.index.as_unit("ns")
 
     tm.assert_equal(test_case, expected_results)
+
+
+def test_holiday_with_exclusion():
+    # GH 54382
+    start = Timestamp("2020-05-01")
+    end = Timestamp("2025-05-31")
+    exclude = [Timestamp("2022-05-30")]  # Queen's platinum Jubilee
+    default_uk_spring_bank_holiday: Holiday = Holiday(
+        "UK Spring Bank Holiday",
+        month=5,
+        day=31,
+        offset=DateOffset(weekday=MO(-1)),
+    )
+
+    queens_jubilee_uk_spring_bank_holiday: Holiday = Holiday(
+        "Queen's Jubilee UK Spring Bank Holiday",
+        month=5,
+        day=31,
+        offset=DateOffset(weekday=MO(-1)),
+        exclude_dates=exclude,
+    )
+
+    original_dates = list(default_uk_spring_bank_holiday.dates(start, end))
+    exclusion_dates = list(queens_jubilee_uk_spring_bank_holiday.dates(start, end))
+
+    assert all(ex in original_dates for ex in exclude)
+    assert all(ex not in exclusion_dates for ex in exclude)
+    assert set(exclusion_dates).issubset(original_dates)
+
+
+def test_holiday_with_multiple_exclusions():
+    start = Timestamp("2000-01-01")
+    end = Timestamp("2100-05-31")
+    exclude = [
+        Timestamp("2025-01-01"),
+        Timestamp("2042-01-01"),
+        Timestamp("2061-01-01"),
+    ]  # Yakudoshi new year
+    default_japan_new_year: Holiday = Holiday(
+        "Japan New Year",
+        month=1,
+        day=1,
+    )
+
+    yakudoshi_new_year: Holiday = Holiday(
+        "Yakudoshi New Year", month=1, day=1, exclude_dates=exclude
+    )
+
+    original_dates = list(default_japan_new_year.dates(start, end))
+    exclusion_dates = list(yakudoshi_new_year.dates(start, end))
+
+    assert all(ex in original_dates for ex in exclude)
+    assert all(ex not in exclusion_dates for ex in exclude)
+    assert set(exclusion_dates).issubset(original_dates)

pandas/tseries/holiday.py

@@ -4,7 +4,10 @@
     datetime,
     timedelta,
 )
-from typing import TYPE_CHECKING
+from typing import (
+    TYPE_CHECKING,
+    Any,
+)
 import warnings
 
 from dateutil.relativedelta import (
@@ -169,6 +172,7 @@ def __init__(
         start_date=None,
         end_date=None,
         days_of_week: tuple | None = None,
+        exclude_dates: list[Any] | None = None,
     ) -> None:
         """
         Parameters
@@ -193,6 +197,9 @@ class from pandas.tseries.offsets, default None
         days_of_week : tuple of int or dateutil.relativedelta weekday strs, default None
             Provide a tuple of days e.g  (0,1,2,3,) for Monday Through Thursday
             Monday=0,..,Sunday=6
+        exclude_dates : list of datetime-likes or
+                        single datetime-like, default None
+            Specific dates to exclude e.g. skipping a specific year's holiday
 
         Examples
         --------
@@ -257,6 +264,11 @@ class from pandas.tseries.offsets, default None
         self.observance = observance
         assert days_of_week is None or type(days_of_week) == tuple
         self.days_of_week = days_of_week
+        self.exclude_dates = (
+            [Timestamp(ex) for ex in exclude_dates]
+            if exclude_dates is not None
+            else exclude_dates
+        )
 
     def __repr__(self) -> str:
         info = ""
@@ -328,6 +340,12 @@ def dates(
         holiday_dates = holiday_dates[
             (holiday_dates >= filter_start_date) & (holiday_dates <= filter_end_date)
         ]
+
+        if self.exclude_dates:
+            holiday_dates = DatetimeIndex(
+                [hd for hd in holiday_dates if hd not in self.exclude_dates]
+            )
+
         if return_name:
             return Series(self.name, index=holiday_dates)
         return holiday_dates