🐴 workaround a mypy/stubtest issue with sys.version_info

Author: jorenham

.mypyignore

@@ -1,7 +1,8 @@
+# intentional type-check-only deviations from runtime
 numpy\.__numtype__
-numpy\.polynomial\._polytypes
 numpy\._typing.*
-numpy\.typing\.mypy_plugin
+numpy\.polynomial\._polytypes
+
 
 # internal testing code
 numpy\.conftest
@@ -25,11 +26,16 @@ numpy\.matlib\.__all__
 # requires numpy/_core/code_generators to be on the PYTHONPATH when running stubtest
 numpy\._core\.cversions
 
+# mypy plugin
+numpy\.typing\.mypy_plugin
+
+
 # workaround mypy's lack of Python 3.13 support for `NamedTuple` types (mypy <= 1.15.0)
-numpy\._core\.overrides\.ArgSpec\.__replace__
-numpy\._utils\._pep440\._Version\.__replace__
-numpy\.lib\._arraysetops_impl\.Unique(All|Counts|Inverse)Result\.__replace__
-numpy\.linalg\._linalg\.(Eig|Eigh|QR|SVD|Slogdet)Result\.__replace__
+numpy(\.\w+)+\.__replace__
 
 # workaround for https://github.com/python/mypy/issues/15302#issuecomment-2701979355
 numpy\.testing(\._private\.utils)?\.check_support_sve
+
+# workaround for stubtest ignoring `if sys.version_info` blocks in `_typeshed` when
+# `python_version`
+numpy\.testing\.TestCase\.[_a-zA-Z]+

.mypyignore-todo

@@ -231,9 +231,4 @@ numpy.polynomial.polynomial.polyvander
 numpy.polynomial.polynomial.polyvander2d
 numpy.polynomial.polynomial.polyvander3d
 
-numpy.testing.TestCase.__init_subclass__
-numpy.testing.TestCase._addSkip
-numpy.testing.TestCase.assertDictContainsSubset
-numpy.testing.TestCase.enter(Class)?Context
-
 numpy.typing.NBitBase

src/numpy-stubs/testing/_private/utils.pyi

@@ -170,10 +170,12 @@ else:
 
 #
 def print_assert_equal(test_string: str, actual: object, desired: object) -> None: ...
+
+#
 def assert_(val: object, msg: str | Callable[[], str] = "") -> None: ...
 def assert_equal(
-    actual: ArrayLike,
-    desired: ArrayLike,
+    actual: object,
+    desired: object,
     err_msg: str = "",
     verbose: bool = True,
     *,
@@ -216,8 +218,8 @@ def assert_array_compare(
 
 #
 def assert_array_equal(
-    actual: ArrayLike,
-    desired: ArrayLike,
+    actual: object,
+    desired: object,
     err_msg: str = "",
     verbose: bool = True,
     *,
@@ -236,26 +238,26 @@ def assert_array_almost_equal(
 #
 @overload
 def assert_array_less(
-    x: _NumericArrayLike,
-    y: _NumericArrayLike,
+    x: _ArrayLikeTD64_co,
+    y: _ArrayLikeTD64_co,
     err_msg: str = "",
     verbose: bool = True,
     *,
     strict: bool = False,
 ) -> None: ...
 @overload
 def assert_array_less(
-    x: _ArrayLikeTD64_co,
-    y: _ArrayLikeTD64_co,
+    x: _ArrayLikeDT64_co,
+    y: _ArrayLikeDT64_co,
     err_msg: str = "",
     verbose: bool = True,
     *,
     strict: bool = False,
 ) -> None: ...
 @overload
 def assert_array_less(
-    x: _ArrayLikeDT64_co,
-    y: _ArrayLikeDT64_co,
+    x: _NumericArrayLike,
+    y: _NumericArrayLike,
     err_msg: str = "",
     verbose: bool = True,
     *,
@@ -302,8 +304,8 @@ def assert_raises_regex(
 #
 @overload
 def assert_allclose(
-    actual: _ArrayLikeNumber_co | _ArrayLikeObject_co,
-    desired: _ArrayLikeNumber_co | _ArrayLikeObject_co,
+    actual: _ArrayLikeTD64_co,
+    desired: _ArrayLikeTD64_co,
     rtol: float = 1e-7,
     atol: float = 0,
     equal_nan: bool = True,
@@ -314,8 +316,8 @@ def assert_allclose(
 ) -> None: ...
 @overload
 def assert_allclose(
-    actual: _ArrayLikeTD64_co,
-    desired: _ArrayLikeTD64_co,
+    actual: _NumericArrayLike,
+    desired: _NumericArrayLike,
     rtol: float = 1e-7,
     atol: float = 0,
     equal_nan: bool = True,

test/static/accept/testing.pyi

@@ -167,8 +167,8 @@ assert_type(np.testing.assert_allclose(AR_i8, AR_f8, verbose=False), None)
 
 assert_type(np.testing.assert_array_almost_equal_nulp(AR_i8, AR_f8, nulp=2), None)
 
-assert_type(np.testing.assert_array_max_ulp(AR_i8, AR_f8, maxulp=2), npt.NDArray[Any])
-assert_type(np.testing.assert_array_max_ulp(AR_i8, AR_f8, dtype=np.float32), npt.NDArray[Any])
+assert_type(np.testing.assert_array_max_ulp(AR_i8, AR_f8, maxulp=2), npt.NDArray[np.floating])
+assert_type(np.testing.assert_array_max_ulp(AR_i8, AR_f8, dtype=np.float32), npt.NDArray[np.floating])
 
 assert_type(np.testing.assert_warns(RuntimeWarning), contextlib._GeneratorContextManager[None])
 assert_type(np.testing.assert_warns(RuntimeWarning, func3, 5), bool)

test/static/reject/testing.pyi

@@ -3,7 +3,7 @@ import numpy.typing as npt
 
 AR_U: npt.NDArray[np.str_]
 
-def func() -> bool: ...
+def func(x: object) -> bool: ...
 
 np.testing.assert_(True, msg=1)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType]  # noqa: PT009
 np.testing.build_err_msg(1, "test")  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType]
@@ -20,9 +20,9 @@ np.testing.assert_allclose(AR_U, AR_U)  # type: ignore[arg-type]  # pyright: ign
 np.testing.assert_array_almost_equal_nulp(AR_U, AR_U)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType]
 np.testing.assert_array_max_ulp(AR_U, AR_U)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType]
 
-np.testing.assert_warns(warning_class=RuntimeWarning, func=func)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
+np.testing.assert_warns(RuntimeWarning, func)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
 np.testing.assert_no_warnings(func=func)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
-np.testing.assert_no_warnings(func, None)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
-np.testing.assert_no_warnings(func, test=None)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
+np.testing.assert_no_warnings(func)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
+np.testing.assert_no_warnings(func, y=None)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]
 
 np.testing.assert_no_gc_cycles(func=func)  # type: ignore[call-overload]  # pyright: ignore[reportCallIssue]