400 lines
9.8 KiB
Python
400 lines
9.8 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import ast
|
||
|
|
import types
|
||
|
|
import warnings
|
||
|
|
import unittest
|
||
|
|
import contextlib
|
||
|
|
from re import Pattern
|
||
|
|
from collections.abc import Callable, Iterable, Sequence
|
||
|
|
from typing import (
|
||
|
|
Literal as L,
|
||
|
|
Any,
|
||
|
|
AnyStr,
|
||
|
|
ClassVar,
|
||
|
|
NoReturn,
|
||
|
|
overload,
|
||
|
|
type_check_only,
|
||
|
|
TypeVar,
|
||
|
|
Union,
|
||
|
|
Final,
|
||
|
|
SupportsIndex,
|
||
|
|
)
|
||
|
|
from typing_extensions import ParamSpec
|
||
|
|
|
||
|
|
from numpy import generic, dtype, number, object_, bool_, _FloatValue
|
||
|
|
from numpy._typing import (
|
||
|
|
NDArray,
|
||
|
|
ArrayLike,
|
||
|
|
DTypeLike,
|
||
|
|
_ArrayLikeNumber_co,
|
||
|
|
_ArrayLikeObject_co,
|
||
|
|
_ArrayLikeTD64_co,
|
||
|
|
_ArrayLikeDT64_co,
|
||
|
|
)
|
||
|
|
|
||
|
|
from unittest.case import (
|
||
|
|
SkipTest as SkipTest,
|
||
|
|
)
|
||
|
|
|
||
|
|
_P = ParamSpec("_P")
|
||
|
|
_T = TypeVar("_T")
|
||
|
|
_ET = TypeVar("_ET", bound=BaseException)
|
||
|
|
_FT = TypeVar("_FT", bound=Callable[..., Any])
|
||
|
|
|
||
|
|
# Must return a bool or an ndarray/generic type
|
||
|
|
# that is supported by `np.logical_and.reduce`
|
||
|
|
_ComparisonFunc = Callable[
|
||
|
|
[NDArray[Any], NDArray[Any]],
|
||
|
|
Union[
|
||
|
|
bool,
|
||
|
|
bool_,
|
||
|
|
number[Any],
|
||
|
|
NDArray[Union[bool_, number[Any], object_]],
|
||
|
|
],
|
||
|
|
]
|
||
|
|
|
||
|
|
__all__: list[str]
|
||
|
|
|
||
|
|
class KnownFailureException(Exception): ...
|
||
|
|
class IgnoreException(Exception): ...
|
||
|
|
|
||
|
|
class clear_and_catch_warnings(warnings.catch_warnings):
|
||
|
|
class_modules: ClassVar[tuple[types.ModuleType, ...]]
|
||
|
|
modules: set[types.ModuleType]
|
||
|
|
@overload
|
||
|
|
def __new__(
|
||
|
|
cls,
|
||
|
|
record: L[False] = ...,
|
||
|
|
modules: Iterable[types.ModuleType] = ...,
|
||
|
|
) -> _clear_and_catch_warnings_without_records: ...
|
||
|
|
@overload
|
||
|
|
def __new__(
|
||
|
|
cls,
|
||
|
|
record: L[True],
|
||
|
|
modules: Iterable[types.ModuleType] = ...,
|
||
|
|
) -> _clear_and_catch_warnings_with_records: ...
|
||
|
|
@overload
|
||
|
|
def __new__(
|
||
|
|
cls,
|
||
|
|
record: bool,
|
||
|
|
modules: Iterable[types.ModuleType] = ...,
|
||
|
|
) -> clear_and_catch_warnings: ...
|
||
|
|
def __enter__(self) -> None | list[warnings.WarningMessage]: ...
|
||
|
|
def __exit__(
|
||
|
|
self,
|
||
|
|
__exc_type: None | type[BaseException] = ...,
|
||
|
|
__exc_val: None | BaseException = ...,
|
||
|
|
__exc_tb: None | types.TracebackType = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
# Type-check only `clear_and_catch_warnings` subclasses for both values of the
|
||
|
|
# `record` parameter. Copied from the stdlib `warnings` stubs.
|
||
|
|
|
||
|
|
@type_check_only
|
||
|
|
class _clear_and_catch_warnings_with_records(clear_and_catch_warnings):
|
||
|
|
def __enter__(self) -> list[warnings.WarningMessage]: ...
|
||
|
|
|
||
|
|
@type_check_only
|
||
|
|
class _clear_and_catch_warnings_without_records(clear_and_catch_warnings):
|
||
|
|
def __enter__(self) -> None: ...
|
||
|
|
|
||
|
|
class suppress_warnings:
|
||
|
|
log: list[warnings.WarningMessage]
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
forwarding_rule: L["always", "module", "once", "location"] = ...,
|
||
|
|
) -> None: ...
|
||
|
|
def filter(
|
||
|
|
self,
|
||
|
|
category: type[Warning] = ...,
|
||
|
|
message: str = ...,
|
||
|
|
module: None | types.ModuleType = ...,
|
||
|
|
) -> None: ...
|
||
|
|
def record(
|
||
|
|
self,
|
||
|
|
category: type[Warning] = ...,
|
||
|
|
message: str = ...,
|
||
|
|
module: None | types.ModuleType = ...,
|
||
|
|
) -> list[warnings.WarningMessage]: ...
|
||
|
|
def __enter__(self: _T) -> _T: ...
|
||
|
|
def __exit__(
|
||
|
|
self,
|
||
|
|
__exc_type: None | type[BaseException] = ...,
|
||
|
|
__exc_val: None | BaseException = ...,
|
||
|
|
__exc_tb: None | types.TracebackType = ...,
|
||
|
|
) -> None: ...
|
||
|
|
def __call__(self, func: _FT) -> _FT: ...
|
||
|
|
|
||
|
|
verbose: int
|
||
|
|
IS_PYPY: Final[bool]
|
||
|
|
IS_PYSTON: Final[bool]
|
||
|
|
HAS_REFCOUNT: Final[bool]
|
||
|
|
HAS_LAPACK64: Final[bool]
|
||
|
|
|
||
|
|
def assert_(val: object, msg: str | Callable[[], str] = ...) -> None: ...
|
||
|
|
|
||
|
|
# Contrary to runtime we can't do `os.name` checks while type checking,
|
||
|
|
# only `sys.platform` checks
|
||
|
|
if sys.platform == "win32" or sys.platform == "cygwin":
|
||
|
|
def memusage(processName: str = ..., instance: int = ...) -> int: ...
|
||
|
|
elif sys.platform == "linux":
|
||
|
|
def memusage(_proc_pid_stat: str | bytes | os.PathLike[Any] = ...) -> None | int: ...
|
||
|
|
else:
|
||
|
|
def memusage() -> NoReturn: ...
|
||
|
|
|
||
|
|
if sys.platform == "linux":
|
||
|
|
def jiffies(
|
||
|
|
_proc_pid_stat: str | bytes | os.PathLike[Any] = ...,
|
||
|
|
_load_time: list[float] = ...,
|
||
|
|
) -> int: ...
|
||
|
|
else:
|
||
|
|
def jiffies(_load_time: list[float] = ...) -> int: ...
|
||
|
|
|
||
|
|
def build_err_msg(
|
||
|
|
arrays: Iterable[object],
|
||
|
|
err_msg: str,
|
||
|
|
header: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
names: Sequence[str] = ...,
|
||
|
|
precision: None | SupportsIndex = ...,
|
||
|
|
) -> str: ...
|
||
|
|
|
||
|
|
def assert_equal(
|
||
|
|
actual: object,
|
||
|
|
desired: object,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def print_assert_equal(
|
||
|
|
test_string: str,
|
||
|
|
actual: object,
|
||
|
|
desired: object,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def assert_almost_equal(
|
||
|
|
actual: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
desired: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
decimal: int = ...,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
# Anything that can be coerced into `builtins.float`
|
||
|
|
def assert_approx_equal(
|
||
|
|
actual: _FloatValue,
|
||
|
|
desired: _FloatValue,
|
||
|
|
significant: int = ...,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def assert_array_compare(
|
||
|
|
comparison: _ComparisonFunc,
|
||
|
|
x: ArrayLike,
|
||
|
|
y: ArrayLike,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
header: str = ...,
|
||
|
|
precision: SupportsIndex = ...,
|
||
|
|
equal_nan: bool = ...,
|
||
|
|
equal_inf: bool = ...,
|
||
|
|
*,
|
||
|
|
strict: bool = ...
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def assert_array_equal(
|
||
|
|
x: ArrayLike,
|
||
|
|
y: ArrayLike,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
*,
|
||
|
|
strict: bool = ...
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def assert_array_almost_equal(
|
||
|
|
x: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
y: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
decimal: float = ...,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_array_less(
|
||
|
|
x: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
y: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
@overload
|
||
|
|
def assert_array_less(
|
||
|
|
x: _ArrayLikeTD64_co,
|
||
|
|
y: _ArrayLikeTD64_co,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
@overload
|
||
|
|
def assert_array_less(
|
||
|
|
x: _ArrayLikeDT64_co,
|
||
|
|
y: _ArrayLikeDT64_co,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def runstring(
|
||
|
|
astr: str | bytes | types.CodeType,
|
||
|
|
dict: None | dict[str, Any],
|
||
|
|
) -> Any: ...
|
||
|
|
|
||
|
|
def assert_string_equal(actual: str, desired: str) -> None: ...
|
||
|
|
|
||
|
|
def rundocs(
|
||
|
|
filename: None | str | os.PathLike[str] = ...,
|
||
|
|
raise_on_error: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def raises(*args: type[BaseException]) -> Callable[[_FT], _FT]: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_raises( # type: ignore
|
||
|
|
expected_exception: type[BaseException] | tuple[type[BaseException], ...],
|
||
|
|
callable: Callable[_P, Any],
|
||
|
|
/,
|
||
|
|
*args: _P.args,
|
||
|
|
**kwargs: _P.kwargs,
|
||
|
|
) -> None: ...
|
||
|
|
@overload
|
||
|
|
def assert_raises(
|
||
|
|
expected_exception: type[_ET] | tuple[type[_ET], ...],
|
||
|
|
*,
|
||
|
|
msg: None | str = ...,
|
||
|
|
) -> unittest.case._AssertRaisesContext[_ET]: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_raises_regex(
|
||
|
|
expected_exception: type[BaseException] | tuple[type[BaseException], ...],
|
||
|
|
expected_regex: str | bytes | Pattern[Any],
|
||
|
|
callable: Callable[_P, Any],
|
||
|
|
/,
|
||
|
|
*args: _P.args,
|
||
|
|
**kwargs: _P.kwargs,
|
||
|
|
) -> None: ...
|
||
|
|
@overload
|
||
|
|
def assert_raises_regex(
|
||
|
|
expected_exception: type[_ET] | tuple[type[_ET], ...],
|
||
|
|
expected_regex: str | bytes | Pattern[Any],
|
||
|
|
*,
|
||
|
|
msg: None | str = ...,
|
||
|
|
) -> unittest.case._AssertRaisesContext[_ET]: ...
|
||
|
|
|
||
|
|
def decorate_methods(
|
||
|
|
cls: type[Any],
|
||
|
|
decorator: Callable[[Callable[..., Any]], Any],
|
||
|
|
testmatch: None | str | bytes | Pattern[Any] = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def measure(
|
||
|
|
code_str: str | bytes | ast.mod | ast.AST,
|
||
|
|
times: int = ...,
|
||
|
|
label: None | str = ...,
|
||
|
|
) -> float: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_allclose(
|
||
|
|
actual: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
desired: _ArrayLikeNumber_co | _ArrayLikeObject_co,
|
||
|
|
rtol: float = ...,
|
||
|
|
atol: float = ...,
|
||
|
|
equal_nan: bool = ...,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
@overload
|
||
|
|
def assert_allclose(
|
||
|
|
actual: _ArrayLikeTD64_co,
|
||
|
|
desired: _ArrayLikeTD64_co,
|
||
|
|
rtol: float = ...,
|
||
|
|
atol: float = ...,
|
||
|
|
equal_nan: bool = ...,
|
||
|
|
err_msg: str = ...,
|
||
|
|
verbose: bool = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def assert_array_almost_equal_nulp(
|
||
|
|
x: _ArrayLikeNumber_co,
|
||
|
|
y: _ArrayLikeNumber_co,
|
||
|
|
nulp: float = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def assert_array_max_ulp(
|
||
|
|
a: _ArrayLikeNumber_co,
|
||
|
|
b: _ArrayLikeNumber_co,
|
||
|
|
maxulp: float = ...,
|
||
|
|
dtype: DTypeLike = ...,
|
||
|
|
) -> NDArray[Any]: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_warns(
|
||
|
|
warning_class: type[Warning],
|
||
|
|
) -> contextlib._GeneratorContextManager[None]: ...
|
||
|
|
@overload
|
||
|
|
def assert_warns(
|
||
|
|
warning_class: type[Warning],
|
||
|
|
func: Callable[_P, _T],
|
||
|
|
/,
|
||
|
|
*args: _P.args,
|
||
|
|
**kwargs: _P.kwargs,
|
||
|
|
) -> _T: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_no_warnings() -> contextlib._GeneratorContextManager[None]: ...
|
||
|
|
@overload
|
||
|
|
def assert_no_warnings(
|
||
|
|
func: Callable[_P, _T],
|
||
|
|
/,
|
||
|
|
*args: _P.args,
|
||
|
|
**kwargs: _P.kwargs,
|
||
|
|
) -> _T: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def tempdir(
|
||
|
|
suffix: None = ...,
|
||
|
|
prefix: None = ...,
|
||
|
|
dir: None = ...,
|
||
|
|
) -> contextlib._GeneratorContextManager[str]: ...
|
||
|
|
@overload
|
||
|
|
def tempdir(
|
||
|
|
suffix: None | AnyStr = ...,
|
||
|
|
prefix: None | AnyStr = ...,
|
||
|
|
dir: None | AnyStr | os.PathLike[AnyStr] = ...,
|
||
|
|
) -> contextlib._GeneratorContextManager[AnyStr]: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def temppath(
|
||
|
|
suffix: None = ...,
|
||
|
|
prefix: None = ...,
|
||
|
|
dir: None = ...,
|
||
|
|
text: bool = ...,
|
||
|
|
) -> contextlib._GeneratorContextManager[str]: ...
|
||
|
|
@overload
|
||
|
|
def temppath(
|
||
|
|
suffix: None | AnyStr = ...,
|
||
|
|
prefix: None | AnyStr = ...,
|
||
|
|
dir: None | AnyStr | os.PathLike[AnyStr] = ...,
|
||
|
|
text: bool = ...,
|
||
|
|
) -> contextlib._GeneratorContextManager[AnyStr]: ...
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def assert_no_gc_cycles() -> contextlib._GeneratorContextManager[None]: ...
|
||
|
|
@overload
|
||
|
|
def assert_no_gc_cycles(
|
||
|
|
func: Callable[_P, Any],
|
||
|
|
/,
|
||
|
|
*args: _P.args,
|
||
|
|
**kwargs: _P.kwargs,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
def break_cycles() -> None: ...
|