GH - 624: date_range() now supports a unit keyword "s" , "ms" , "us" , or "ns" (#734)

* date_range() now supports a unit keyword "s" , "ms" , "us" , or "ns"

* fix problems

* numpy = "<=1.24.3"

* pyright = "<= 1.1.313"

* black && isort

Author: asukaminato0721

pandas-stubs/core/indexes/datetimes.pyi

@@ -6,7 +6,10 @@ from datetime import (
     timedelta,
     tzinfo,
 )
-from typing import overload
+from typing import (
+    Literal,
+    overload,
+)
 
 import numpy as np
 from pandas import (
@@ -104,6 +107,7 @@ def date_range(
     normalize: bool = ...,
     name: Hashable | None = ...,
     inclusive: IntervalClosedType = ...,
+    unit: Literal["s", "ms", "us", "ns"] | None = ...,
 ) -> DatetimeIndex: ...
 @overload
 def bdate_range(

pyproject.toml

@@ -33,14 +33,14 @@ packages = [
 [tool.poetry.dependencies]
 python = ">=3.8"
 types-pytz = ">= 2022.1.1"
-numpy = ">=1.24.3"
+numpy = "<=1.24.3"
 
 [tool.poetry.dev-dependencies]
 mypy = "1.3.0"
 pandas = "2.0.2"
 pyarrow = ">=10.0.1"
 pytest = ">=7.1.2"
-pyright = ">= 1.1.310"
+pyright = "<= 1.1.313"
 poethepoet = ">=0.16.5"
 loguru = ">=0.6.0"
 typing-extensions = ">=4.4.0"

tests/test_timefuncs.py

@@ -1177,3 +1177,34 @@ def test_weekofmonth_init():
         ),
         pd.offsets.WeekOfMonth,
     )
+
+
+def test_date_range_unit():
+    check(
+        assert_type(
+            pd.date_range("1/1/2022", "2/1/2022", unit="s"),
+            pd.DatetimeIndex,
+        ),
+        pd.DatetimeIndex,
+    )
+    check(
+        assert_type(
+            pd.date_range("1/1/2022", "2/1/2022", unit="ms"),
+            pd.DatetimeIndex,
+        ),
+        pd.DatetimeIndex,
+    )
+    check(
+        assert_type(
+            pd.date_range("1/1/2022", "2/1/2022", unit="us"),
+            pd.DatetimeIndex,
+        ),
+        pd.DatetimeIndex,
+    )
+    check(
+        assert_type(
+            pd.date_range("1/1/2022", "2/1/2022", unit="ns"),
+            pd.DatetimeIndex,
+        ),
+        pd.DatetimeIndex,
+    )