diff --git a/pandas-stubs/core/series.pyi b/pandas-stubs/core/series.pyi index 149d90fb8..12f8f9524 100644 --- a/pandas-stubs/core/series.pyi +++ b/pandas-stubs/core/series.pyi @@ -360,7 +360,7 @@ class Series(IndexOpsMixin[S1], NDFrame): | tuple[Hashable | slice, ...], ) -> Self: ... @overload - def __getitem__(self, idx: int | _str) -> S1: ... + def __getitem__(self, idx: Scalar) -> S1: ... def __setitem__(self, key, value) -> None: ... def repeat( self, repeats: int | list[int], axis: AxisIndex | None = ... diff --git a/tests/test_series.py b/tests/test_series.py index 43fdd4afb..e5262fc79 100644 --- a/tests/test_series.py +++ b/tests/test_series.py @@ -999,6 +999,12 @@ def test_types_getitem() -> None: s3: pd.Series = s[:2] +def test_types_getitem_by_timestamp() -> None: + index = pd.date_range("2018-01-01", periods=2, freq="D") + series = pd.Series(range(2), index=index) + check(assert_type(series[index[-1]], int), np.integer) + + def test_types_eq() -> None: s1 = pd.Series([1, 2, 3]) res1: pd.Series = s1 == 1