22
22
_SUNDAY = datetime (2014 , 4 , 13 )
23
23
_MONDAY = datetime (2014 , 4 , 14 )
24
24
_TUESDAY = datetime (2014 , 4 , 15 )
25
+ _NEXT_WEDNESDAY = datetime (2014 , 4 , 16 )
25
26
26
27
27
28
@pytest .mark .parametrize ("day" , [_SATURDAY , _SUNDAY ])
@@ -60,7 +61,15 @@ def test_weekend_to_monday(day, expected):
60
61
61
62
62
63
@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
+ ],
64
73
)
65
74
def test_next_workday (day , expected ):
66
75
assert next_workday (day ) == expected
@@ -74,7 +83,16 @@ def test_previous_workday(day, expected):
74
83
75
84
76
85
@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
+ ],
78
96
)
79
97
def test_before_nearest_workday (day , expected ):
80
98
assert before_nearest_workday (day ) == expected
0 commit comments