GH648 Allowing dateoffset weekday from relativedelta

Author: loicdiridollou

pandas-stubs/_libs/tslibs/offsets.pyi

@@ -12,6 +12,7 @@ from typing import (
     overload,
 )
 
+from dateutil.relativedelta import weekday as WeekdayClass
 import numpy as np
 from pandas.core.indexes.datetimes import DatetimeIndex
 from typing_extensions import Self
@@ -257,7 +258,7 @@ class DateOffset(RelativeDeltaOffset):
         year: int = ...,
         month: int = ...,
         day: int = ...,
-        weekday: int = ...,
+        weekday: int | WeekdayClass = ...,
         hour: int = ...,
         minute: int = ...,
         second: int = ...,

tests/test_timefuncs.py

@@ -8,6 +8,15 @@
     cast,
 )
 
+from dateutil.relativedelta import (
+    FR,
+    MO,
+    SA,
+    SU,
+    TH,
+    TU,
+    WE,
+)
 import numpy as np
 from numpy import typing as npt
 import pandas as pd
@@ -19,7 +28,12 @@
 
 from pandas._libs import NaTType
 from pandas._libs.tslibs import BaseOffset
-from pandas._libs.tslibs.offsets import DateOffset
+from pandas._libs.tslibs.offsets import (
+    DateOffset,
+    FY5253Mixin,
+    WeekOfMonth,
+    WeekOfMonthMixin,
+)
 
 if TYPE_CHECKING:
     from pandas._typing import FulldatetimeDict
@@ -1284,6 +1298,21 @@ def test_weekofmonth_init():
     )
 
 
+def test_dateoffset_weekday():
+    """Check that you can create a `pd.DateOffset` from weekday of int or dateutil.relativedelta."""
+    # check for int
+    check(
+        assert_type(pd.offsets.DateOffset(weekday=1), pd.offsets.DateOffset),
+        pd.offsets.DateOffset,
+    )
+    # check for relativedelta
+    for weekday in [MO, TU, WE, TH, TH, FR, SA, SU]:
+        check(
+            assert_type(pd.offsets.DateOffset(weekday=weekday), pd.offsets.DateOffset),
+            pd.offsets.DateOffset,
+        )
+
+
 def test_date_range_unit():
     check(
         assert_type(