145 lines
5.5 KiB
Python
145 lines
5.5 KiB
Python
|
import datetime as dt
|
||
|
|
from typing import Any, TypeVar
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import numpy.typing as npt
|
||
|
|
|
||
|
|
_SCT = TypeVar("_SCT", bound=np.generic, covariant=True)
|
||
|
|
|
||
|
|
class SubClass(np.ndarray[Any, np.dtype[_SCT]]): ...
|
||
|
|
|
||
|
|
subclass: SubClass[np.float64]
|
||
|
|
|
||
|
|
AR_f8: npt.NDArray[np.float64]
|
||
|
|
AR_i8: npt.NDArray[np.int64]
|
||
|
|
AR_u1: npt.NDArray[np.uint8]
|
||
|
|
AR_m: npt.NDArray[np.timedelta64]
|
||
|
|
AR_M: npt.NDArray[np.datetime64]
|
||
|
|
|
||
|
|
AR_LIKE_f: list[float]
|
||
|
|
AR_LIKE_i: list[int]
|
||
|
|
|
||
|
|
m: np.timedelta64
|
||
|
|
M: np.datetime64
|
||
|
|
|
||
|
|
b_f8 = np.broadcast(AR_f8)
|
||
|
|
b_i8_f8_f8 = np.broadcast(AR_i8, AR_f8, AR_f8)
|
||
|
|
|
||
|
|
nditer_obj: np.nditer
|
||
|
|
|
||
|
|
date_scalar: dt.date
|
||
|
|
date_seq: list[dt.date]
|
||
|
|
timedelta_seq: list[dt.timedelta]
|
||
|
|
|
||
|
|
def func(a: int) -> bool: ...
|
||
|
|
|
||
|
|
reveal_type(next(b_f8)) # E: tuple[Any, ...]
|
||
|
|
reveal_type(b_f8.reset()) # E: None
|
||
|
|
reveal_type(b_f8.index) # E: int
|
||
|
|
reveal_type(b_f8.iters) # E: tuple[flatiter[Any], ...]
|
||
|
|
reveal_type(b_f8.nd) # E: int
|
||
|
|
reveal_type(b_f8.ndim) # E: int
|
||
|
|
reveal_type(b_f8.numiter) # E: int
|
||
|
|
reveal_type(b_f8.shape) # E: tuple[builtins.int, ...]
|
||
|
|
reveal_type(b_f8.size) # E: int
|
||
|
|
|
||
|
|
reveal_type(next(b_i8_f8_f8)) # E: tuple[Any, ...]
|
||
|
|
reveal_type(b_i8_f8_f8.reset()) # E: None
|
||
|
|
reveal_type(b_i8_f8_f8.index) # E: int
|
||
|
|
reveal_type(b_i8_f8_f8.iters) # E: tuple[flatiter[Any], ...]
|
||
|
|
reveal_type(b_i8_f8_f8.nd) # E: int
|
||
|
|
reveal_type(b_i8_f8_f8.ndim) # E: int
|
||
|
|
reveal_type(b_i8_f8_f8.numiter) # E: int
|
||
|
|
reveal_type(b_i8_f8_f8.shape) # E: tuple[builtins.int, ...]
|
||
|
|
reveal_type(b_i8_f8_f8.size) # E: int
|
||
|
|
|
||
|
|
reveal_type(np.inner(AR_f8, AR_i8)) # E: Any
|
||
|
|
|
||
|
|
reveal_type(np.where([True, True, False])) # E: tuple[ndarray[Any, dtype[{intp}]], ...]
|
||
|
|
reveal_type(np.where([True, True, False], 1, 0)) # E: ndarray[Any, dtype[Any]]
|
||
|
|
|
||
|
|
reveal_type(np.lexsort([0, 1, 2])) # E: Any
|
||
|
|
|
||
|
|
reveal_type(np.can_cast(np.dtype("i8"), int)) # E: bool
|
||
|
|
reveal_type(np.can_cast(AR_f8, "f8")) # E: bool
|
||
|
|
reveal_type(np.can_cast(AR_f8, np.complex128, casting="unsafe")) # E: bool
|
||
|
|
|
||
|
|
reveal_type(np.min_scalar_type([1])) # E: dtype[Any]
|
||
|
|
reveal_type(np.min_scalar_type(AR_f8)) # E: dtype[Any]
|
||
|
|
|
||
|
|
reveal_type(np.result_type(int, [1])) # E: dtype[Any]
|
||
|
|
reveal_type(np.result_type(AR_f8, AR_u1)) # E: dtype[Any]
|
||
|
|
reveal_type(np.result_type(AR_f8, np.complex128)) # E: dtype[Any]
|
||
|
|
|
||
|
|
reveal_type(np.dot(AR_LIKE_f, AR_i8)) # E: Any
|
||
|
|
reveal_type(np.dot(AR_u1, 1)) # E: Any
|
||
|
|
reveal_type(np.dot(1.5j, 1)) # E: Any
|
||
|
|
reveal_type(np.dot(AR_u1, 1, out=AR_f8)) # E: ndarray[Any, dtype[{float64}]]
|
||
|
|
|
||
|
|
reveal_type(np.vdot(AR_LIKE_f, AR_i8)) # E: floating[Any]
|
||
|
|
reveal_type(np.vdot(AR_u1, 1)) # E: signedinteger[Any]
|
||
|
|
reveal_type(np.vdot(1.5j, 1)) # E: complexfloating[Any, Any]
|
||
|
|
|
||
|
|
reveal_type(np.bincount(AR_i8)) # E: ndarray[Any, dtype[{intp}]]
|
||
|
|
|
||
|
|
reveal_type(np.copyto(AR_f8, [1., 1.5, 1.6])) # E: None
|
||
|
|
|
||
|
|
reveal_type(np.putmask(AR_f8, [True, True, False], 1.5)) # E: None
|
||
|
|
|
||
|
|
reveal_type(np.packbits(AR_i8)) # ndarray[Any, dtype[{uint8}]]
|
||
|
|
reveal_type(np.packbits(AR_u1)) # ndarray[Any, dtype[{uint8}]]
|
||
|
|
|
||
|
|
reveal_type(np.unpackbits(AR_u1)) # ndarray[Any, dtype[{uint8}]]
|
||
|
|
|
||
|
|
reveal_type(np.shares_memory(1, 2)) # E: bool
|
||
|
|
reveal_type(np.shares_memory(AR_f8, AR_f8, max_work=1)) # E: bool
|
||
|
|
|
||
|
|
reveal_type(np.may_share_memory(1, 2)) # E: bool
|
||
|
|
reveal_type(np.may_share_memory(AR_f8, AR_f8, max_work=1)) # E: bool
|
||
|
|
|
||
|
|
reveal_type(np.geterrobj()) # E: list[Any]
|
||
|
|
|
||
|
|
reveal_type(np.seterrobj([8192, 521, None])) # E: None
|
||
|
|
|
||
|
|
reveal_type(np.promote_types(np.int32, np.int64)) # E: dtype[Any]
|
||
|
|
reveal_type(np.promote_types("f4", float)) # E: dtype[Any]
|
||
|
|
|
||
|
|
reveal_type(np.frompyfunc(func, 1, 1, identity=None)) # ufunc
|
||
|
|
|
||
|
|
reveal_type(np.datetime_data("m8[D]")) # E: Tuple[builtins.str, builtins.int]
|
||
|
|
reveal_type(np.datetime_data(np.datetime64)) # E: Tuple[builtins.str, builtins.int]
|
||
|
|
reveal_type(np.datetime_data(np.dtype(np.timedelta64))) # E: Tuple[builtins.str, builtins.int]
|
||
|
|
|
||
|
|
reveal_type(np.busday_count("2011-01", "2011-02")) # E: {int_}
|
||
|
|
reveal_type(np.busday_count(["2011-01"], "2011-02")) # E: ndarray[Any, dtype[{int_}]]
|
||
|
|
reveal_type(np.busday_count(["2011-01"], date_scalar)) # E: ndarray[Any, dtype[{int_}]]
|
||
|
|
|
||
|
|
reveal_type(np.busday_offset(M, m)) # E: datetime64
|
||
|
|
reveal_type(np.busday_offset(date_scalar, m)) # E: datetime64
|
||
|
|
reveal_type(np.busday_offset(M, 5)) # E: datetime64
|
||
|
|
reveal_type(np.busday_offset(AR_M, m)) # E: ndarray[Any, dtype[datetime64]]
|
||
|
|
reveal_type(np.busday_offset(M, timedelta_seq)) # E: ndarray[Any, dtype[datetime64]]
|
||
|
|
reveal_type(np.busday_offset("2011-01", "2011-02", roll="forward")) # E: datetime64
|
||
|
|
reveal_type(np.busday_offset(["2011-01"], "2011-02", roll="forward")) # E: ndarray[Any, dtype[datetime64]]
|
||
|
|
|
||
|
|
reveal_type(np.is_busday("2012")) # E: bool_
|
||
|
|
reveal_type(np.is_busday(date_scalar)) # E: bool_
|
||
|
|
reveal_type(np.is_busday(["2012"])) # E: ndarray[Any, dtype[bool_]]
|
||
|
|
|
||
|
|
reveal_type(np.datetime_as_string(M)) # E: str_
|
||
|
|
reveal_type(np.datetime_as_string(AR_M)) # E: ndarray[Any, dtype[str_]]
|
||
|
|
|
||
|
|
reveal_type(np.busdaycalendar(holidays=date_seq)) # E: busdaycalendar
|
||
|
|
reveal_type(np.busdaycalendar(holidays=[M])) # E: busdaycalendar
|
||
|
|
|
||
|
|
reveal_type(np.compare_chararrays("a", "b", "!=", rstrip=False)) # E: ndarray[Any, dtype[bool_]]
|
||
|
|
reveal_type(np.compare_chararrays(b"a", b"a", "==", True)) # E: ndarray[Any, dtype[bool_]]
|
||
|
|
|
||
|
|
reveal_type(np.add_docstring(func, "test")) # E: None
|
||
|
|
|
||
|
|
reveal_type(np.nested_iters([AR_i8, AR_i8], [[0], [1]], flags=["c_index"])) # E: tuple[nditer, ...]
|
||
|
|
reveal_type(np.nested_iters([AR_i8, AR_i8], [[0], [1]], op_flags=[["readonly", "readonly"]])) # E: tuple[nditer, ...]
|
||
|
|
reveal_type(np.nested_iters([AR_i8, AR_i8], [[0], [1]], op_dtypes=np.int_)) # E: tuple[nditer, ...]
|
||
|
|
reveal_type(np.nested_iters([AR_i8, AR_i8], [[0], [1]], order="C", casting="no")) # E: tuple[nditer, ...]
|