File tree 2 files changed +47
-1
lines changed
pandas-stubs/_libs/tslibs
2 files changed +47
-1
lines changed Original file line number Diff line number Diff line change @@ -12,6 +12,7 @@ from typing import (
12
12
overload ,
13
13
)
14
14
15
+ from dateutil .relativedelta import weekday as WeekdayClass
15
16
import numpy as np
16
17
from pandas .core .indexes .datetimes import DatetimeIndex
17
18
from typing_extensions import Self
@@ -257,7 +258,7 @@ class DateOffset(RelativeDeltaOffset):
257
258
year : int = ...,
258
259
month : int = ...,
259
260
day : int = ...,
260
- weekday : int = ...,
261
+ weekday : int | WeekdayClass = ...,
261
262
hour : int = ...,
262
263
minute : int = ...,
263
264
second : int = ...,
Original file line number Diff line number Diff line change 8
8
cast ,
9
9
)
10
10
11
+ from dateutil .relativedelta import (
12
+ FR ,
13
+ MO ,
14
+ SA ,
15
+ SU ,
16
+ TH ,
17
+ TU ,
18
+ WE ,
19
+ )
11
20
import numpy as np
12
21
from numpy import typing as npt
13
22
import pandas as pd
@@ -1284,6 +1293,42 @@ def test_weekofmonth_init():
1284
1293
)
1285
1294
1286
1295
1296
+ def test_dateoffset_weekday () -> None :
1297
+ """Check that you can create a `pd.DateOffset` from weekday of int or relativedelta.weekday."""
1298
+ check (
1299
+ assert_type (pd .offsets .DateOffset (weekday = 1 ), pd .offsets .DateOffset ),
1300
+ pd .offsets .DateOffset ,
1301
+ )
1302
+ check (
1303
+ assert_type (pd .offsets .DateOffset (weekday = MO ), pd .offsets .DateOffset ),
1304
+ pd .offsets .DateOffset ,
1305
+ )
1306
+ check (
1307
+ assert_type (pd .offsets .DateOffset (weekday = TU ), pd .offsets .DateOffset ),
1308
+ pd .offsets .DateOffset ,
1309
+ )
1310
+ check (
1311
+ assert_type (pd .offsets .DateOffset (weekday = WE ), pd .offsets .DateOffset ),
1312
+ pd .offsets .DateOffset ,
1313
+ )
1314
+ check (
1315
+ assert_type (pd .offsets .DateOffset (weekday = TH ), pd .offsets .DateOffset ),
1316
+ pd .offsets .DateOffset ,
1317
+ )
1318
+ check (
1319
+ assert_type (pd .offsets .DateOffset (weekday = FR ), pd .offsets .DateOffset ),
1320
+ pd .offsets .DateOffset ,
1321
+ )
1322
+ check (
1323
+ assert_type (pd .offsets .DateOffset (weekday = SA ), pd .offsets .DateOffset ),
1324
+ pd .offsets .DateOffset ,
1325
+ )
1326
+ check (
1327
+ assert_type (pd .offsets .DateOffset (weekday = SU ), pd .offsets .DateOffset ),
1328
+ pd .offsets .DateOffset ,
1329
+ )
1330
+
1331
+
1287
1332
def test_date_range_unit ():
1288
1333
check (
1289
1334
assert_type (
You can’t perform that action at this time.
0 commit comments