462 lines
11 KiB
Python
462 lines
11 KiB
Python
"""
|
|
This type stub file was generated by pyright.
|
|
"""
|
|
|
|
from typing import Any, NoReturn, Protocol, TypeVar, overload
|
|
from numpy import bool_, complex128, complexfloating, float64, floating, generic, int8, int_, integer, number, signedinteger, timedelta64, unsignedinteger
|
|
from ._nbit import _NBitDouble, _NBitInt
|
|
from ._scalars import _BoolLike_co, _FloatLike_co, _IntLike_co, _NumberLike_co
|
|
from . import NBitBase
|
|
from ._array_like import NDArray
|
|
from ._nested_sequence import _NestedSequence
|
|
|
|
"""
|
|
A module with various ``typing.Protocol`` subclasses that implement
|
|
the ``__call__`` magic method.
|
|
|
|
See the `Mypy documentation`_ on protocols for more details.
|
|
|
|
.. _`Mypy documentation`: https://mypy.readthedocs.io/en/stable/protocols.html#callback-protocols
|
|
|
|
"""
|
|
_T1 = TypeVar("_T1")
|
|
_T2 = TypeVar("_T2")
|
|
_T1_contra = TypeVar("_T1_contra", contravariant=True)
|
|
_T2_contra = TypeVar("_T2_contra", contravariant=True)
|
|
_2Tuple = tuple[_T1, _T1]
|
|
_NBit1 = TypeVar("_NBit1", bound=NBitBase)
|
|
_NBit2 = TypeVar("_NBit2", bound=NBitBase)
|
|
_IntType = TypeVar("_IntType", bound=integer)
|
|
_FloatType = TypeVar("_FloatType", bound=floating)
|
|
_NumberType = TypeVar("_NumberType", bound=number)
|
|
_NumberType_co = TypeVar("_NumberType_co", covariant=True, bound=number)
|
|
_GenericType_co = TypeVar("_GenericType_co", covariant=True, bound=generic)
|
|
class _BoolOp(Protocol[_GenericType_co]):
|
|
@overload
|
|
def __call__(self, other: _BoolLike_co, /) -> _GenericType_co:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> int_:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> float64:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complex128:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _NumberType, /) -> _NumberType:
|
|
...
|
|
|
|
|
|
|
|
class _BoolBitOp(Protocol[_GenericType_co]):
|
|
@overload
|
|
def __call__(self, other: _BoolLike_co, /) -> _GenericType_co:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> int_:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _IntType, /) -> _IntType:
|
|
...
|
|
|
|
|
|
|
|
class _BoolSub(Protocol):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> NoReturn:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> int_:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> float64:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complex128:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _NumberType, /) -> _NumberType:
|
|
...
|
|
|
|
|
|
|
|
class _BoolTrueDiv(Protocol):
|
|
@overload
|
|
def __call__(self, other: float | _IntLike_co, /) -> float64:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complex128:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _NumberType, /) -> _NumberType:
|
|
...
|
|
|
|
|
|
|
|
class _BoolMod(Protocol):
|
|
@overload
|
|
def __call__(self, other: _BoolLike_co, /) -> int8:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> int_:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> float64:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _IntType, /) -> _IntType:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _FloatType, /) -> _FloatType:
|
|
...
|
|
|
|
|
|
|
|
class _BoolDivMod(Protocol):
|
|
@overload
|
|
def __call__(self, other: _BoolLike_co, /) -> _2Tuple[int8]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> _2Tuple[int_]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> _2Tuple[floating[_NBit1 | _NBitDouble]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _IntType, /) -> _2Tuple[_IntType]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _FloatType, /) -> _2Tuple[_FloatType]:
|
|
...
|
|
|
|
|
|
|
|
class _TD64Div(Protocol[_NumberType_co]):
|
|
@overload
|
|
def __call__(self, other: timedelta64, /) -> _NumberType_co:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _BoolLike_co, /) -> NoReturn:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _FloatLike_co, /) -> timedelta64:
|
|
...
|
|
|
|
|
|
|
|
class _IntTrueDiv(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> floating[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> floating[_NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complexfloating[_NBit1 | _NBitDouble, _NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: integer[_NBit2], /) -> floating[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _UnsignedIntOp(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> unsignedinteger[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int | signedinteger[Any], /) -> Any:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complexfloating[_NBit1 | _NBitDouble, _NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: unsignedinteger[_NBit2], /) -> unsignedinteger[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _UnsignedIntBitOp(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> unsignedinteger[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> signedinteger[Any]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: signedinteger[Any], /) -> signedinteger[Any]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: unsignedinteger[_NBit2], /) -> unsignedinteger[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _UnsignedIntMod(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> unsignedinteger[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int | signedinteger[Any], /) -> Any:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: unsignedinteger[_NBit2], /) -> unsignedinteger[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _UnsignedIntDivMod(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> _2Tuple[signedinteger[_NBit1]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int | signedinteger[Any], /) -> _2Tuple[Any]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> _2Tuple[floating[_NBit1 | _NBitDouble]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: unsignedinteger[_NBit2], /) -> _2Tuple[unsignedinteger[_NBit1 | _NBit2]]:
|
|
...
|
|
|
|
|
|
|
|
class _SignedIntOp(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> signedinteger[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> signedinteger[_NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complexfloating[_NBit1 | _NBitDouble, _NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: signedinteger[_NBit2], /) -> signedinteger[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _SignedIntBitOp(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> signedinteger[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> signedinteger[_NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: signedinteger[_NBit2], /) -> signedinteger[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _SignedIntMod(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> signedinteger[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> signedinteger[_NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: signedinteger[_NBit2], /) -> signedinteger[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _SignedIntDivMod(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> _2Tuple[signedinteger[_NBit1]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> _2Tuple[signedinteger[_NBit1 | _NBitInt]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> _2Tuple[floating[_NBit1 | _NBitDouble]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: signedinteger[_NBit2], /) -> _2Tuple[signedinteger[_NBit1 | _NBit2]]:
|
|
...
|
|
|
|
|
|
|
|
class _FloatOp(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> floating[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> floating[_NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complexfloating[_NBit1 | _NBitDouble, _NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: integer[_NBit2] | floating[_NBit2], /) -> floating[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _FloatMod(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> floating[_NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> floating[_NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> floating[_NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: integer[_NBit2] | floating[_NBit2], /) -> floating[_NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _FloatDivMod(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> _2Tuple[floating[_NBit1]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> _2Tuple[floating[_NBit1 | _NBitInt]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: float, /) -> _2Tuple[floating[_NBit1 | _NBitDouble]]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: integer[_NBit2] | floating[_NBit2], /) -> _2Tuple[floating[_NBit1 | _NBit2]]:
|
|
...
|
|
|
|
|
|
|
|
class _ComplexOp(Protocol[_NBit1]):
|
|
@overload
|
|
def __call__(self, other: bool, /) -> complexfloating[_NBit1, _NBit1]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: int, /) -> complexfloating[_NBit1 | _NBitInt, _NBit1 | _NBitInt]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: complex, /) -> complexfloating[_NBit1 | _NBitDouble, _NBit1 | _NBitDouble]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: (integer[_NBit2] | floating[_NBit2] | complexfloating[_NBit2, _NBit2]), /) -> complexfloating[_NBit1 | _NBit2, _NBit1 | _NBit2]:
|
|
...
|
|
|
|
|
|
|
|
class _NumberOp(Protocol):
|
|
def __call__(self, other: _NumberLike_co, /) -> Any:
|
|
...
|
|
|
|
|
|
|
|
class _SupportsLT(Protocol):
|
|
def __lt__(self, other: Any, /) -> object:
|
|
...
|
|
|
|
|
|
|
|
class _SupportsGT(Protocol):
|
|
def __gt__(self, other: Any, /) -> object:
|
|
...
|
|
|
|
|
|
|
|
class _ComparisonOp(Protocol[_T1_contra, _T2_contra]):
|
|
@overload
|
|
def __call__(self, other: _T1_contra, /) -> bool_:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _T2_contra, /) -> NDArray[bool_]:
|
|
...
|
|
|
|
@overload
|
|
def __call__(self, other: _SupportsLT | _SupportsGT | _NestedSequence[_SupportsLT | _SupportsGT], /) -> Any:
|
|
...
|
|
|
|
|
|
|