70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
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
|