Initial configuration commit

This commit is contained in:
Alex Selimov 2023-10-24 22:54:55 -04:00
commit 31c8abea59
266 changed files with 780274 additions and 0 deletions

View file

@ -0,0 +1,11 @@
"""
This type stub file was generated by pyright.
"""
from numpy._pytesttester import PytestTester
from unittest import TestCase as TestCase
from numpy.testing._private.utils import HAS_LAPACK64 as HAS_LAPACK64, HAS_REFCOUNT as HAS_REFCOUNT, IS_PYPY as IS_PYPY, IS_PYSTON as IS_PYSTON, IgnoreException as IgnoreException, KnownFailureException as KnownFailureException, SkipTest as SkipTest, assert_ as assert_, assert_allclose as assert_allclose, assert_almost_equal as assert_almost_equal, assert_approx_equal as assert_approx_equal, assert_array_almost_equal as assert_array_almost_equal, assert_array_almost_equal_nulp as assert_array_almost_equal_nulp, assert_array_compare as assert_array_compare, assert_array_equal as assert_array_equal, assert_array_less as assert_array_less, assert_array_max_ulp as assert_array_max_ulp, assert_equal as assert_equal, assert_no_gc_cycles as assert_no_gc_cycles, assert_no_warnings as assert_no_warnings, assert_raises as assert_raises, assert_raises_regex as assert_raises_regex, assert_string_equal as assert_string_equal, assert_warns as assert_warns, break_cycles as break_cycles, build_err_msg as build_err_msg, clear_and_catch_warnings as clear_and_catch_warnings, decorate_methods as decorate_methods, jiffies as jiffies, measure as measure, memusage as memusage, print_assert_equal as print_assert_equal, rundocs as rundocs, runstring as runstring, suppress_warnings as suppress_warnings, tempdir as tempdir, temppath as temppath, verbose as verbose
__all__: list[str]
__path__: list[str]
test: PytestTester

View file

@ -0,0 +1,241 @@
"""
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:
...