fix: np.ndarray // pd.IndexOpsMixin

Author: cmp0xff

tests/indexes/arithmetic/float/test_floordiv.py

@@ -92,9 +92,11 @@ def test_floordiv_numpy_array(left: "pd.Index[float]") -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rfloordiv__` cannot override. At runtime, they lead to
     # errors or pd.Index.
-    check(b // left, pd.Index, np.floating)
-    check(i // left, pd.Index, np.floating)
-    check(f // left, pd.Index, np.floating)
+    check(assert_type(b // left, "np.typing.NDArray[np.int8]"), pd.Index, np.floating)
+    check(assert_type(i // left, "np.typing.NDArray[np.int64]"), pd.Index, np.floating)
+    check(
+        assert_type(f // left, "np.typing.NDArray[np.float64]"), pd.Index, np.floating
+    )
     if TYPE_CHECKING_INVALID_USAGE:
         assert_type(c // left, Any)
         assert_type(s // left, Any)

tests/indexes/arithmetic/int/test_floordiv.py

@@ -92,13 +92,19 @@ def test_floordiv_numpy_array(left: "pd.Index[int]") -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rfloordiv__` cannot override. At runtime, they lead to
     # errors or pd.Index.
-    check(b // left, pd.Index, np.integer)
-    check(i // left, pd.Index, np.integer)
-    check(f // left, pd.Index, np.floating)
+    check(assert_type(b // left, "np.typing.NDArray[np.int8]"), pd.Index, np.integer)
+    check(assert_type(i // left, "np.typing.NDArray[np.int64]"), pd.Index, np.integer)
+    check(
+        assert_type(f // left, "np.typing.NDArray[np.float64]"), pd.Index, np.floating
+    )
     if TYPE_CHECKING_INVALID_USAGE:
         assert_type(c // left, Any)
         assert_type(s // left, Any)
-        assert_type(d // left, "np.typing.NDArray[np.int64]")
+    check(
+        assert_type(d // left, "np.typing.NDArray[np.int64]"),
+        pd.TimedeltaIndex,
+        pd.Timedelta,
+    )
 
 
 def test_floordiv_pd_index(left: "pd.Index[int]") -> None:

tests/indexes/arithmetic/test_floordiv.py

@@ -92,13 +92,19 @@ def test_floordiv_numpy_array(left_i: pd.Index) -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rfloordiv__` cannot override. At runtime, they lead to
     # errors or pd.Index.
-    check(b // left_i, pd.Index, np.integer)
-    check(i // left_i, pd.Index, np.integer)
-    check(f // left_i, pd.Index, np.floating)
+    check(assert_type(b // left_i, "np.typing.NDArray[np.int8]"), pd.Index, np.integer)
+    check(assert_type(i // left_i, "np.typing.NDArray[np.int64]"), pd.Index, np.integer)
+    check(
+        assert_type(f // left_i, "np.typing.NDArray[np.float64]"), pd.Index, np.floating
+    )
     if TYPE_CHECKING_INVALID_USAGE:
         assert_type(c // left_i, Any)
         assert_type(s // left_i, Any)
-        assert_type(d // left_i, Any)  # pyright: ignore[reportAssertTypeFailure]
+    check(
+        assert_type(d // left_i, "np.typing.NDArray[np.int64]"),
+        pd.TimedeltaIndex,
+        pd.Timedelta,
+    )
 
 
 def test_floordiv_pd_index(left_i: pd.Index) -> None:

tests/series/arithmetic/int/test_floordiv.py

@@ -132,13 +132,17 @@ def test_floordiv_numpy_array(left: "pd.Series[int]") -> None:
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rfloordiv__` cannot override. At runtime, they lead to
     # errors or pd.Series.
-    check(b // left, pd.Series, np.integer)
-    check(i // left, pd.Series, np.integer)
-    check(f // left, pd.Series, np.floating)
+    check(assert_type(b // left, "np.typing.NDArray[np.int8]"), pd.Series, np.integer)
+    check(assert_type(i // left, "np.typing.NDArray[np.int64]"), pd.Series, np.integer)
+    check(
+        assert_type(f // left, "np.typing.NDArray[np.float64]"), pd.Series, np.floating
+    )
     if TYPE_CHECKING_INVALID_USAGE:
         assert_type(c // left, Any)
         assert_type(s // left, Any)
-        assert_type(d // left, "np.typing.NDArray[np.int64]")
+    check(
+        assert_type(d // left, "np.typing.NDArray[np.int64]"), pd.Series, pd.Timedelta
+    )
 
     check(assert_type(left.floordiv(b), "pd.Series[int]"), pd.Series, np.integer)
     check(assert_type(left.floordiv(i), "pd.Series[int]"), pd.Series, np.integer)