fixup

Author: MarcoGorelli

pandas-stubs/_libs/missing.pyi

@@ -66,7 +66,6 @@ class NAType:
     def __rmul__(self, other: Index, /) -> Index: ...  # type: ignore[overload-overlap]
     @overload
     def __rmul__(self, other: object, /) -> NAType: ...
-    @overload
     def __matmul__(self, other: object, /) -> NAType: ...
     def __rmatmul__(self, other: object, /) -> NAType: ...
     @overload

tests/test_natype.py

@@ -91,25 +91,29 @@ def test_arithmetic() -> None:
     with pytest.raises(RuntimeError):
         # bug upstream: https://github.com/pandas-dev/pandas/issues/62196
         check(
-            assert_type(divmod(na, s_int), tuple[pd.Series, pd.Series]), tuple
-        )  # pyright: ignore[reportAssertTypeFailure]
+            assert_type(divmod(na, s_int), tuple[pd.Series, pd.Series]),
+            tuple,  # pyright: ignore[reportAssertTypeFailure]
+        )
     with pytest.raises(RuntimeError):
         check(
-            assert_type(divmod(na, idx_int), tuple[pd.Index, pd.Index]), tuple
-        )  # pyright: ignore[reportAssertTypeFailure]
+            assert_type(divmod(na, idx_int), tuple[pd.Index, pd.Index]),
+            tuple,  # pyright: ignore[reportAssertTypeFailure]
+        )
     check(assert_type(divmod(na, 1), tuple[NAType, NAType]), tuple)
 
     # __rdivmod__
     with pytest.raises(RuntimeError):
         # bug upstream: https://github.com/pandas-dev/pandas/issues/62196
         check(
-            assert_type(divmod(s_int, na), tuple[pd.Series, pd.Series]), tuple
-        )  # pyright: ignore[reportAssertTypeFailure]
+            assert_type(divmod(s_int, na), tuple[pd.Series, pd.Series]),
+            tuple,  # pyright: ignore[reportAssertTypeFailure]
+        )
     with pytest.raises(RuntimeError):
         # https://github.com/pandas-dev/pandas-stubs/issues/1347
         check(
-            assert_type(divmod(idx_int, na), tuple[pd.Index, pd.Index]), tuple
-        )  # pyright: ignore[reportAssertTypeFailure]
+            assert_type(divmod(idx_int, na), tuple[pd.Index, pd.Index]),
+            tuple,  # pyright: ignore[reportAssertTypeFailure]
+        )
     check(assert_type(divmod(1, na), tuple[NAType, NAType]), tuple)
 
     # __eq__
@@ -132,16 +136,16 @@ def test_arithmetic() -> None:
     check(assert_type(na < idx_int, BooleanArray), BooleanArray)
     check(assert_type(na < 1, NAType), NAType)
 
-    # __ge__
-    check(assert_type(na >= s_int, pd.Series), pd.Series)
-    check(assert_type(na >= idx_int, BooleanArray), BooleanArray)
-    check(assert_type(na >= 1, NAType), NAType)
-
     # __gt__
     check(assert_type(na > s_int, pd.Series), pd.Series)
     check(assert_type(na > idx_int, BooleanArray), BooleanArray)
     check(assert_type(na > 1, NAType), NAType)
 
+    # __ge__
+    check(assert_type(na >= s_int, pd.Series), pd.Series)
+    check(assert_type(na >= idx_int, BooleanArray), BooleanArray)
+    check(assert_type(na >= 1, NAType), NAType)
+
     # __pow__
     check(assert_type(na**s_int, pd.Series), pd.Series)
     check(assert_type(na**idx_int, pd.Index), pd.Index)