codekingpro/portable-devtools

1import numpy as np
2from numpy._typing import NDArray
3from numpy.testing import (
4    TestCase,
5    assert_,
6    assert_allclose,
7    assert_array_almost_equal_nulp,
8    assert_raises,
9)
10from numpy.testing._private.utils import _ComparisonFunc
11
12__all__ = [
13    "TestCase",
14    "almost",
15    "approx",
16    "assert_",
17    "assert_allclose",
18    "assert_almost_equal",
19    "assert_array_almost_equal",
20    "assert_array_almost_equal_nulp",
21    "assert_array_approx_equal",
22    "assert_array_compare",
23    "assert_array_equal",
24    "assert_array_less",
25    "assert_close",
26    "assert_equal",
27    "assert_equal_records",
28    "assert_mask_equal",
29    "assert_not_equal",
30    "assert_raises",
31    "fail_if_array_equal",
32]
33
34def approx(
35    a: object, b: object, fill_value: bool = True, rtol: float = 1e-5, atol: float = 1e-8
36) -> np.ndarray[tuple[int], np.dtype[np.bool]]: ...
37def almost(a: object, b: object, decimal: int = 6, fill_value: bool = True) -> np.ndarray[tuple[int], np.dtype[np.bool]]: ...
38
39#
40def assert_equal_records(a: NDArray[np.void], b: NDArray[np.void]) -> None: ...
41def assert_equal(actual: object, desired: object, err_msg: str = "") -> None: ...
42def fail_if_equal(actual: object, desired: object, err_msg: str = "") -> None: ...
43def assert_almost_equal(
44    actual: object, desired: object, decimal: int = 7, err_msg: str = "", verbose: bool = True
45) -> None: ...
46
47#
48def assert_array_compare(
49    comparison: _ComparisonFunc,
50    x: object,
51    y: object,
52    err_msg: str = "",
53    verbose: bool = True,
54    header: str = "",
55    fill_value: bool = True,
56) -> None: ...
57def assert_array_equal(x: object, y: object, err_msg: str = "", verbose: bool = True) -> None: ...
58def fail_if_array_equal(x: object, y: object, err_msg: str = "", verbose: bool = True) -> None: ...
59def assert_array_approx_equal(
60    x: object, y: object, decimal: int = 6, err_msg: str = "", verbose: bool = True
61) -> None: ...
62def assert_array_almost_equal(
63    x: object, y: object, decimal: int = 6, err_msg: str = "", verbose: bool = True
64) -> None: ...
65def assert_array_less(x: object, y: object, err_msg: str = "", verbose: bool = True) -> None: ...
66def assert_mask_equal(m1: object, m2: object, err_msg: str = "") -> None: ...
67
68assert_not_equal = fail_if_equal
69assert_close = assert_almost_equal
70