make ShapeT work for eq and ne

Author: Dr-Irv

pandas-stubs/_libs/tslibs/timestamps.pyi

@@ -45,7 +45,6 @@ from pandas._typing import (
     TimeUnit,
     np_1darray,
     np_ndarray,
-    npt,
 )
 
 _Ambiguous: TypeAlias = bool | Literal["raise", "NaT"]
@@ -260,7 +259,7 @@ class Timestamp(datetime, SupportsIndex):
     @overload
     def __eq__(self, other: Index) -> np_1darray[np.bool]: ...  # type: ignore[overload-overlap]
     @overload  # TODO: using shape-aware arrays similar to other methods doesn't work in mypy
-    def __eq__(self, other: npt.NDArray[np.datetime64]) -> npt.NDArray[np.bool]: ...  # type: ignore[overload-overlap]
+    def __eq__(self, other: np_ndarray[ShapeT, np.datetime64]) -> np_ndarray[ShapeT, np.bool]: ...  # type: ignore[overload-overlap]
     @overload
     def __eq__(self, other: object) -> Literal[False]: ...
     @overload
@@ -270,7 +269,7 @@ class Timestamp(datetime, SupportsIndex):
     @overload
     def __ne__(self, other: Index) -> np_1darray[np.bool]: ...  # type: ignore[overload-overlap]
     @overload  # TODO: using shape-aware arrays similar to other methods doesn't work in mypy
-    def __ne__(self, other: npt.NDArray[np.datetime64]) -> npt.NDArray[np.bool]: ...  # type: ignore[overload-overlap]
+    def __ne__(self, other: np_ndarray[ShapeT, np.datetime64]) -> np_ndarray[ShapeT, np.bool]: ...  # type: ignore[overload-overlap]
     @overload
     def __ne__(self, other: object) -> Literal[True]: ...
     def __hash__(self) -> int: ...

tests/test_scalars.py

@@ -1394,20 +1394,28 @@ def test_timestamp_cmp() -> None:
         # tests in this block fail with mypy on Python 3.10 in CI only
         # I couldn't reproduce the failure locally so skip mypy on Python 3.10
         eq1_arr = check(
-            assert_type(ts == c_np_ndarray_dt64, np_ndarray_bool), np.ndarray, np.bool_
+            assert_type(ts == c_np_ndarray_dt64, np_ndarray_bool),
+            np_1darray[np.bool],
         )
         ne1_arr = check(
             assert_type(ts != c_np_ndarray_dt64, np_ndarray_bool), np.ndarray, np.bool_
         )
         assert (eq1_arr != ne1_arr).all()
         # TODO: the following should be 2D-arrays but it doesn't work in mypy
-        eq1_arr = check(
-            assert_type(ts == c_np_2darray_dt64, np_ndarray_bool), np_ndarray_bool
+
+        eq2_arr = check(
+            assert_type(
+                ts == c_np_2darray_dt64, np.ndarray[tuple[int, int], np.dtype[np.bool]]
+            ),
+            np_ndarray_bool,
         )
-        ne1_arr = check(
-            assert_type(ts != c_np_2darray_dt64, np_ndarray_bool), np_ndarray_bool
+        ne2_arr = check(
+            assert_type(
+                ts != c_np_2darray_dt64, np.ndarray[tuple[int, int], np.dtype[np.bool]]
+            ),
+            np_ndarray_bool,
         )
-        assert (eq1_arr != ne1_arr).all()
+        assert (eq2_arr != ne2_arr).all()
 
     eq_s = check(
         assert_type(ts == c_series_timestamp, "pd.Series[bool]"), pd.Series, np.bool_