| | import abc |
| | import decimal |
| | from collections.abc import Iterator, Sequence |
| | from typing import ( |
| | Any, |
| | ClassVar, |
| | Generic, |
| | Literal, |
| | Self, |
| | SupportsIndex, |
| | TypeAlias, |
| | overload, |
| | ) |
| | from typing_extensions import TypeIs, TypeVar |
| |
|
| | import numpy as np |
| | import numpy.typing as npt |
| | from numpy._typing import ( |
| | _ArrayLikeComplex_co, |
| | _ArrayLikeFloat_co, |
| | _FloatLike_co, |
| | _NumberLike_co, |
| | ) |
| |
|
| | from ._polytypes import ( |
| | _AnyInt, |
| | _Array2, |
| | _ArrayLikeCoef_co, |
| | _ArrayLikeCoefObject_co, |
| | _CoefLike_co, |
| | _CoefSeries, |
| | _Series, |
| | _SeriesLikeCoef_co, |
| | _SeriesLikeInt_co, |
| | _Tuple2, |
| | ) |
| |
|
| | __all__ = ["ABCPolyBase"] |
| |
|
| | _NameT_co = TypeVar("_NameT_co", bound=str | None, default=str | None, covariant=True) |
| | _PolyT = TypeVar("_PolyT", bound=ABCPolyBase) |
| | _AnyOther: TypeAlias = ABCPolyBase | _CoefLike_co | _SeriesLikeCoef_co |
| |
|
| | class ABCPolyBase(Generic[_NameT_co], abc.ABC): |
| | __hash__: ClassVar[None] = None |
| | __array_ufunc__: ClassVar[None] = None |
| | maxpower: ClassVar[Literal[100]] = 100 |
| |
|
| | _superscript_mapping: ClassVar[dict[int, str]] = ... |
| | _subscript_mapping: ClassVar[dict[int, str]] = ... |
| | _use_unicode: ClassVar[bool] = ... |
| |
|
| | _symbol: str |
| | @property |
| | def symbol(self, /) -> str: ... |
| | @property |
| | @abc.abstractmethod |
| | def domain(self) -> _Array2[np.float64 | Any]: ... |
| | @property |
| | @abc.abstractmethod |
| | def window(self) -> _Array2[np.float64 | Any]: ... |
| | @property |
| | @abc.abstractmethod |
| | def basis_name(self) -> _NameT_co: ... |
| |
|
| | coef: _CoefSeries |
| |
|
| | def __init__( |
| | self, |
| | /, |
| | coef: _SeriesLikeCoef_co, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> None: ... |
| |
|
| | |
| | @overload |
| | def __call__(self, /, arg: _PolyT) -> _PolyT: ... |
| | @overload |
| | def __call__(self, /, arg: _FloatLike_co | decimal.Decimal) -> np.float64 | Any: ... |
| | @overload |
| | def __call__(self, /, arg: _NumberLike_co) -> np.complex128 | Any: ... |
| | @overload |
| | def __call__(self, /, arg: _ArrayLikeFloat_co) -> npt.NDArray[np.float64 | Any]: ... |
| | @overload |
| | def __call__(self, /, arg: _ArrayLikeComplex_co) -> npt.NDArray[np.complex128 | Any]: ... |
| | @overload |
| | def __call__(self, /, arg: _ArrayLikeCoefObject_co) -> npt.NDArray[np.object_]: ... |
| |
|
| | |
| | def __neg__(self, /) -> Self: ... |
| | def __pos__(self, /) -> Self: ... |
| |
|
| | |
| | def __add__(self, x: _AnyOther, /) -> Self: ... |
| | def __sub__(self, x: _AnyOther, /) -> Self: ... |
| | def __mul__(self, x: _AnyOther, /) -> Self: ... |
| | def __pow__(self, x: _AnyOther, /) -> Self: ... |
| | def __truediv__(self, x: _AnyOther, /) -> Self: ... |
| | def __floordiv__(self, x: _AnyOther, /) -> Self: ... |
| | def __mod__(self, x: _AnyOther, /) -> Self: ... |
| | def __divmod__(self, x: _AnyOther, /) -> _Tuple2[Self]: ... |
| |
|
| | |
| | def __radd__(self, x: _AnyOther, /) -> Self: ... |
| | def __rsub__(self, x: _AnyOther, /) -> Self: ... |
| | def __rmul__(self, x: _AnyOther, /) -> Self: ... |
| | def __rtruediv__(self, x: _AnyOther, /) -> Self: ... |
| | def __rfloordiv__(self, x: _AnyOther, /) -> Self: ... |
| | def __rmod__(self, x: _AnyOther, /) -> Self: ... |
| | def __rdivmod__(self, x: _AnyOther, /) -> _Tuple2[Self]: ... |
| |
|
| | |
| | def __len__(self, /) -> int: ... |
| | def __iter__(self, /) -> Iterator[np.float64 | Any]: ... |
| |
|
| | |
| | def __getstate__(self, /) -> dict[str, Any]: ... |
| | def __setstate__(self, dict: dict[str, Any], /) -> None: ... |
| |
|
| | |
| | def has_samecoef(self, /, other: ABCPolyBase) -> bool: ... |
| | def has_samedomain(self, /, other: ABCPolyBase) -> bool: ... |
| | def has_samewindow(self, /, other: ABCPolyBase) -> bool: ... |
| | def has_sametype(self, /, other: object) -> TypeIs[Self]: ... |
| |
|
| | |
| | def copy(self, /) -> Self: ... |
| | def degree(self, /) -> int: ... |
| | def cutdeg(self, /, deg: int) -> Self: ... |
| | def trim(self, /, tol: _FloatLike_co = 0) -> Self: ... |
| | def truncate(self, /, size: _AnyInt) -> Self: ... |
| |
|
| | |
| | @overload |
| | def convert( |
| | self, |
| | /, |
| | domain: _SeriesLikeCoef_co | None, |
| | kind: type[_PolyT], |
| | window: _SeriesLikeCoef_co | None = None, |
| | ) -> _PolyT: ... |
| | @overload |
| | def convert( |
| | self, |
| | /, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | *, |
| | kind: type[_PolyT], |
| | window: _SeriesLikeCoef_co | None = None, |
| | ) -> _PolyT: ... |
| | @overload |
| | def convert( |
| | self, |
| | /, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | kind: None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | ) -> Self: ... |
| |
|
| | |
| | def mapparms(self, /) -> _Tuple2[Any]: ... |
| | def integ( |
| | self, |
| | /, |
| | m: SupportsIndex = 1, |
| | k: _CoefLike_co | _SeriesLikeCoef_co = [], |
| | lbnd: _CoefLike_co | None = None, |
| | ) -> Self: ... |
| | def deriv(self, /, m: SupportsIndex = 1) -> Self: ... |
| | def roots(self, /) -> _CoefSeries: ... |
| | def linspace( |
| | self, |
| | /, |
| | n: SupportsIndex = 100, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | ) -> _Tuple2[_Series[np.float64 | np.complex128]]: ... |
| |
|
| | |
| | @overload |
| | @classmethod |
| | def fit( |
| | cls, |
| | x: _SeriesLikeCoef_co, |
| | y: _SeriesLikeCoef_co, |
| | deg: int | _SeriesLikeInt_co, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | rcond: _FloatLike_co | None = None, |
| | full: Literal[False] = False, |
| | w: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> Self: ... |
| | @overload |
| | @classmethod |
| | def fit( |
| | cls, |
| | x: _SeriesLikeCoef_co, |
| | y: _SeriesLikeCoef_co, |
| | deg: int | _SeriesLikeInt_co, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | rcond: _FloatLike_co | None = None, |
| | *, |
| | full: Literal[True], |
| | w: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> tuple[Self, Sequence[np.inexact | np.int32]]: ... |
| | @overload |
| | @classmethod |
| | def fit( |
| | cls, |
| | x: _SeriesLikeCoef_co, |
| | y: _SeriesLikeCoef_co, |
| | deg: int | _SeriesLikeInt_co, |
| | domain: _SeriesLikeCoef_co | None, |
| | rcond: _FloatLike_co, |
| | full: Literal[True], |
| | /, |
| | w: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> tuple[Self, Sequence[np.inexact | np.int32]]: ... |
| |
|
| | |
| | @classmethod |
| | def fromroots( |
| | cls, |
| | roots: _ArrayLikeCoef_co, |
| | domain: _SeriesLikeCoef_co | None = [], |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> Self: ... |
| | @classmethod |
| | def identity( |
| | cls, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> Self: ... |
| | @classmethod |
| | def basis( |
| | cls, |
| | deg: _AnyInt, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | symbol: str = "x", |
| | ) -> Self: ... |
| | @classmethod |
| | def cast( |
| | cls, |
| | series: ABCPolyBase, |
| | domain: _SeriesLikeCoef_co | None = None, |
| | window: _SeriesLikeCoef_co | None = None, |
| | ) -> Self: ... |
| | @classmethod |
| | def _str_term_unicode(cls, /, i: str, arg_str: str) -> str: ... |
| | @classmethod |
| | def _str_term_ascii(cls, /, i: str, arg_str: str) -> str: ... |
| | @classmethod |
| | def _repr_latex_term(cls, /, i: str, arg_str: str, needs_parens: bool) -> str: ... |
| |
|
