diff --git a/pandas-stubs/core/indexes/datetimes.pyi b/pandas-stubs/core/indexes/datetimes.pyi index 86d549318..615cb098e 100644 --- a/pandas-stubs/core/indexes/datetimes.pyi +++ b/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( diff --git a/pyproject.toml b/pyproject.toml index 79e5b342c..6eec09492 100644 --- a/pyproject.toml +++ b/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" diff --git a/tests/test_timefuncs.py b/tests/test_timefuncs.py index 20a965545..34d37ac9f 100644 --- a/tests/test_timefuncs.py +++ b/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, + )