Skip to content

Commit 6001e4b

Browse files
GH648 Clean up testing
1 parent 27b6024 commit 6001e4b

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

tests/test_timefuncs.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,12 @@
1616
TH,
1717
TU,
1818
WE,
19+
weekday as WeekdayClass,
1920
)
2021
import numpy as np
2122
from numpy import typing as npt
2223
import pandas as pd
24+
import pytest
2325
import pytz
2426
from typing_extensions import (
2527
assert_never,
@@ -1293,19 +1295,13 @@ def test_weekofmonth_init():
12931295
)
12941296

12951297

1296-
def test_dateoffset_weekday():
1297-
"""Check that you can create a `pd.DateOffset` from weekday of int or dateutil.relativedelta."""
1298-
# check for int
1298+
@pytest.mark.parametrize("weekday", [1, MO, TU, WE, TH, TH, FR, SA, SU])
1299+
def test_dateoffset_weekday(weekday: int | WeekdayClass):
1300+
"""Check that you can create a `pd.DateOffset` from weekday of int or relativedelta.weekday."""
12991301
check(
1300-
assert_type(pd.offsets.DateOffset(weekday=1), pd.offsets.DateOffset),
1302+
assert_type(pd.offsets.DateOffset(weekday=weekday), pd.offsets.DateOffset),
13011303
pd.offsets.DateOffset,
13021304
)
1303-
# check for relativedelta
1304-
for weekday in [MO, TU, WE, TH, TH, FR, SA, SU]:
1305-
check(
1306-
assert_type(pd.offsets.DateOffset(weekday=weekday), pd.offsets.DateOffset),
1307-
pd.offsets.DateOffset,
1308-
)
13091305

13101306

13111307
def test_date_range_unit():

0 commit comments

Comments
 (0)