Added Timestamp to Series#__getitem__ (#775)

* Added `Timestamp` to `Series#__getitem__`

#773

* Fixed code review comments

* Replaced `int | _str` with `Scalar`

* Removed `Timestamp` (it's included in `Scalar`)

* Fixed failing tests by replacing `int` with `np.integer`

Author: GarrisonD

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

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