1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
| from datetime import datetime
|
| from pandas import DatetimeIndex
| import pandas._testing as tm
|
| from pandas.tseries.holiday import (
| AbstractHolidayCalendar,
| USFederalHolidayCalendar,
| USMartinLutherKingJr,
| USMemorialDay,
| )
|
|
| def test_no_mlk_before_1986():
| # see gh-10278
| class MLKCalendar(AbstractHolidayCalendar):
| rules = [USMartinLutherKingJr]
|
| holidays = MLKCalendar().holidays(start="1984", end="1988").to_pydatetime().tolist()
|
| # Testing to make sure holiday is not incorrectly observed before 1986.
| assert holidays == [datetime(1986, 1, 20, 0, 0), datetime(1987, 1, 19, 0, 0)]
|
|
| def test_memorial_day():
| class MemorialDay(AbstractHolidayCalendar):
| rules = [USMemorialDay]
|
| holidays = MemorialDay().holidays(start="1971", end="1980").to_pydatetime().tolist()
|
| # Fixes 5/31 error and checked manually against Wikipedia.
| assert holidays == [
| datetime(1971, 5, 31, 0, 0),
| datetime(1972, 5, 29, 0, 0),
| datetime(1973, 5, 28, 0, 0),
| datetime(1974, 5, 27, 0, 0),
| datetime(1975, 5, 26, 0, 0),
| datetime(1976, 5, 31, 0, 0),
| datetime(1977, 5, 30, 0, 0),
| datetime(1978, 5, 29, 0, 0),
| datetime(1979, 5, 28, 0, 0),
| ]
|
|
| def test_federal_holiday_inconsistent_returntype():
| # GH 49075 test case
| # Instantiate two calendars to rule out _cache
| cal1 = USFederalHolidayCalendar()
| cal2 = USFederalHolidayCalendar()
|
| results_2018 = cal1.holidays(start=datetime(2018, 8, 1), end=datetime(2018, 8, 31))
| results_2019 = cal2.holidays(start=datetime(2019, 8, 1), end=datetime(2019, 8, 31))
| expected_results = DatetimeIndex([], dtype="datetime64[ns]", freq=None)
|
| # Check against expected results to ensure both date
| # ranges generate expected results as per GH49075 submission
| tm.assert_index_equal(results_2018, expected_results)
| tm.assert_index_equal(results_2019, expected_results)
|
|
