Skip to content

Commit 7757844

Browse files
authored
TST: add cases for nearest_workday and before_nearest_workday (#38215)
1 parent a48957b commit 7757844

File tree

1 file changed

+20
-2
lines changed

1 file changed

+20
-2
lines changed

pandas/tests/tseries/holiday/test_observance.py

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
_SUNDAY = datetime(2014, 4, 13)
2323
_MONDAY = datetime(2014, 4, 14)
2424
_TUESDAY = datetime(2014, 4, 15)
25+
_NEXT_WEDNESDAY = datetime(2014, 4, 16)
2526

2627

2728
@pytest.mark.parametrize("day", [_SATURDAY, _SUNDAY])
@@ -60,7 +61,15 @@ def test_weekend_to_monday(day, expected):
6061

6162

6263
@pytest.mark.parametrize(
63-
"day,expected", [(_SATURDAY, _MONDAY), (_SUNDAY, _MONDAY), (_MONDAY, _TUESDAY)]
64+
"day,expected",
65+
[
66+
(_WEDNESDAY, _THURSDAY),
67+
(_THURSDAY, _FRIDAY),
68+
(_SATURDAY, _MONDAY),
69+
(_SUNDAY, _MONDAY),
70+
(_MONDAY, _TUESDAY),
71+
(_TUESDAY, _NEXT_WEDNESDAY), # WED is same week as TUE
72+
],
6473
)
6574
def test_next_workday(day, expected):
6675
assert next_workday(day) == expected
@@ -74,7 +83,16 @@ def test_previous_workday(day, expected):
7483

7584

7685
@pytest.mark.parametrize(
77-
"day,expected", [(_SATURDAY, _THURSDAY), (_SUNDAY, _FRIDAY), (_TUESDAY, _MONDAY)]
86+
"day,expected",
87+
[
88+
(_THURSDAY, _WEDNESDAY),
89+
(_FRIDAY, _THURSDAY),
90+
(_SATURDAY, _THURSDAY),
91+
(_SUNDAY, _FRIDAY),
92+
(_MONDAY, _FRIDAY), # last week Friday
93+
(_TUESDAY, _MONDAY),
94+
(_NEXT_WEDNESDAY, _TUESDAY), # WED is same week as TUE
95+
],
7896
)
7997
def test_before_nearest_workday(day, expected):
8098
assert before_nearest_workday(day) == expected

0 commit comments

Comments
 (0)