Add timedelta as possible freq parameter type to date_range() and bdate_range()

Author: gandhis1

pandas-stubs/core/indexes/datetimes.pyi

@@ -1,4 +1,4 @@
-from datetime import tzinfo
+from datetime import timedelta, tzinfo
 from typing import overload
 
 import numpy as np
@@ -90,7 +90,7 @@ def date_range(
     start: str | DatetimeLike | None = ...,
     end: str | DatetimeLike | None = ...,
     periods: int | None = ...,
-    freq: str | BaseOffset = ...,
+    freq: str | timedelta | BaseOffset = ...,
     tz: str | tzinfo = ...,
     normalize: bool = ...,
     name: str | None = ...,
@@ -101,7 +101,7 @@ def bdate_range(
     start: str | DatetimeLike | None = ...,
     end: str | DatetimeLike | None = ...,
     periods: int | None = ...,
-    freq: str | BaseOffset = ...,
+    freq: str | timedelta | BaseOffset = ...,
     tz: str | tzinfo = ...,
     normalize: bool = ...,
     name: str | None = ...,

tests/test_timefuncs.py

@@ -458,7 +458,6 @@ def test_timedeltaindex_accessors() -> None:
 
 def test_some_offsets() -> None:
     # GH 222
-
     check(
         assert_type(
             CustomBusinessDay(calendar=USFederalHolidayCalendar()), CustomBusinessDay
@@ -484,6 +483,19 @@ def test_some_offsets() -> None:
     )
     # GH 224
     check(assert_type(dt.date.today() - Day(), dt.date), dt.date)
+    # GH 235
+    check(
+        assert_type(
+            pd.date_range("1/1/2022", "2/1/2022", freq=dt.timedelta(days=2)), pd.DatetimeIndex
+        ),
+        pd.DatetimeIndex,
+    )
+    check(
+        assert_type(
+            pd.bdate_range("1/1/2022", "2/1/2022", freq=dt.timedelta(days=2)), pd.DatetimeIndex
+        ),
+        pd.DatetimeIndex,
+    )
 
 
 def test_types_to_numpy() -> None: