|
|
import numpy as np |
|
|
from numpy._typing import NDArray |
|
|
from numpy.testing import ( |
|
|
TestCase, |
|
|
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: NDArray[np.void], b: 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 |
|
|
|
