241 lines
8.6 KiB
Python
241 lines
8.6 KiB
Python
"""
|
|
This type stub file was generated by pyright.
|
|
"""
|
|
|
|
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 Any, AnyStr, ClassVar, Final, Literal as L, ParamSpec, SupportsIndex, TypeVar, Union, overload, type_check_only
|
|
from numpy import _FloatValue, bool_, number, object_
|
|
from numpy._typing import ArrayLike, DTypeLike, NDArray, _ArrayLikeDT64_co, _ArrayLikeNumber_co, _ArrayLikeObject_co, _ArrayLikeTD64_co
|
|
|
|
if sys.version_info >= (3, 10):
|
|
...
|
|
else:
|
|
...
|
|
_P = ParamSpec("_P")
|
|
_T = TypeVar("_T")
|
|
_ET = TypeVar("_ET", bound=BaseException)
|
|
_FT = TypeVar("_FT", bound=Callable[..., Any])
|
|
_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
|
|
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:
|
|
...
|
|
|
|
if sys.platform == "win32" or sys.platform == "cygwin":
|
|
...
|
|
else:
|
|
def memusage(_proc_pid_stat: str | bytes | os.PathLike[Any] = ...) -> None | int:
|
|
...
|
|
|
|
if sys.platform == "linux":
|
|
def jiffies(_proc_pid_stat: str | bytes | os.PathLike[Any] = ..., _load_time: list[float] = ...) -> int:
|
|
...
|
|
|
|
else:
|
|
...
|
|
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:
|
|
...
|
|
|
|
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(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:
|
|
...
|
|
|