Merge pull request #281 from numpy/ma.testutils

🏷️ stub `numpy.ma.testutils`

Author: jorenham

src/numpy-stubs/ma/testutils.pyi

@@ -0,0 +1,71 @@
+from unittest import TestCase as _TestCase
+
+import numpy as np
+import numpy.typing as npt
+from numpy.testing import assert_, assert_allclose, assert_array_almost_equal_nulp, assert_raises
+from numpy.testing._private.utils import _ComparisonFunc
+
+__all__ = [
+    "TestCase",
+    "almost",
+    "approx",
+    "assert_",
+    "assert_allclose",
+    "assert_almost_equal",
+    "assert_array_almost_equal",
+    "assert_array_almost_equal_nulp",
+    "assert_array_approx_equal",
+    "assert_array_compare",
+    "assert_array_equal",
+    "assert_array_less",
+    "assert_close",
+    "assert_equal",
+    "assert_equal_records",
+    "assert_mask_equal",
+    "assert_not_equal",
+    "assert_raises",
+    "fail_if_array_equal",
+]
+
+def approx(
+    a: object,
+    b: object,
+    fill_value: bool = True,
+    rtol: float = 1e-5,
+    atol: float = 1e-8,
+) -> np.ndarray[tuple[int], np.dtype[np.bool]]: ...
+def almost(
+    a: object,
+    b: object,
+    decimal: int = 6,
+    fill_value: bool = True,
+) -> np.ndarray[tuple[int], np.dtype[np.bool]]: ...
+
+#
+def assert_equal_records(a: npt.NDArray[np.void], b: npt.NDArray[np.void]) -> None: ...
+def assert_equal(actual: object, desired: object, err_msg: str = "") -> None: ...
+def fail_if_equal(actual: object, desired: object, err_msg: str = "") -> None: ...
+def assert_almost_equal(actual: object, desired: object, decimal: int = 7, err_msg: str = "", verbose: bool = True) -> None: ...
+
+#
+def assert_array_compare(
+    comparison: _ComparisonFunc,
+    x: object,
+    y: object,
+    err_msg: str = "",
+    verbose: bool = True,
+    header: str = "",
+    fill_value: bool = True,
+) -> None: ...
+def assert_array_equal(x: object, y: object, err_msg: str = "", verbose: bool = True) -> None: ...
+def fail_if_array_equal(x: object, y: object, err_msg: str = "", verbose: bool = True) -> None: ...
+def assert_array_approx_equal(x: object, y: object, decimal: int = 6, err_msg: str = "", verbose: bool = True) -> None: ...
+def assert_array_almost_equal(x: object, y: object, decimal: int = 6, err_msg: str = "", verbose: bool = True) -> None: ...
+def assert_array_less(x: object, y: object, err_msg: str = "", verbose: bool = True) -> None: ...
+def assert_mask_equal(m1: object, m2: object, err_msg: str = "") -> None: ...
+
+assert_not_equal = fail_if_equal
+assert_close = assert_almost_equal
+
+class TestCase(_TestCase):
+    def __init_subclass__(cls, *args: object, **kwargs: object) -> None: ...

tool/.mypyignore-todo.txt

@@ -20,5 +20,3 @@ numpy.f2py.diagnose
 numpy.f2py.crackfortran
 numpy.f2py.symbolic
 numpy.f2py.((cb|common|f90mod|use)_)?rules
-
-numpy.ma.testutils