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 = ...,

pandas-stubs/core/series.pyi

@@ -1,5 +1,6 @@
 from datetime import (
     date,
+    datetime,
     time,
 )
 from typing import (
@@ -1218,7 +1219,9 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
     def __rtruediv__(self, other: num | _ListLike | Series[S1]) -> Series: ...
     def __rxor__(self, other: num | _ListLike | Series[S1]) -> Series[_bool]: ...
     @overload
-    def __sub__(self, other: Timestamp | TimestampSeries) -> TimedeltaSeries: ...
+    def __sub__(
+        self, other: Timestamp | datetime | TimestampSeries
+    ) -> TimedeltaSeries: ...
     @overload
     def __sub__(
         self, other: Timedelta | TimedeltaSeries | TimedeltaIndex

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: