관리-도구

편집 파일: ndarray_misc.pyi

""" Tests for miscellaneous (non-magic) ``np.ndarray``/``np.generic`` methods. More extensive tests are performed for the methods' function-based counterpart in `../from_numeric.py`. """ import operator import ctypes as ct from typing import Any import numpy as np from numpy._typing import NDArray class SubClass(NDArray[np.object_]): ... f8: np.float64 B: SubClass AR_f8: NDArray[np.float64] AR_i8: NDArray[np.int64] AR_U: NDArray[np.str_] AR_V: NDArray[np.void] ctypes_obj = AR_f8.ctypes reveal_type(AR_f8.__dlpack__()) # E: Any reveal_type(AR_f8.__dlpack_device__()) # E: Tuple[int, Literal[0]] reveal_type(ctypes_obj.data) # E: int reveal_type(ctypes_obj.shape) # E: ctypes.Array[{c_intp}] reveal_type(ctypes_obj.strides) # E: ctypes.Array[{c_intp}] reveal_type(ctypes_obj._as_parameter_) # E: ctypes.c_void_p reveal_type(ctypes_obj.data_as(ct.c_void_p)) # E: ctypes.c_void_p reveal_type(ctypes_obj.shape_as(ct.c_longlong)) # E: ctypes.Array[ctypes.c_longlong] reveal_type(ctypes_obj.strides_as(ct.c_ubyte)) # E: ctypes.Array[ctypes.c_ubyte] reveal_type(f8.all()) # E: bool_ reveal_type(AR_f8.all()) # E: bool_ reveal_type(AR_f8.all(axis=0)) # E: Any reveal_type(AR_f8.all(keepdims=True)) # E: Any reveal_type(AR_f8.all(out=B)) # E: SubClass reveal_type(f8.any()) # E: bool_ reveal_type(AR_f8.any()) # E: bool_ reveal_type(AR_f8.any(axis=0)) # E: Any reveal_type(AR_f8.any(keepdims=True)) # E: Any reveal_type(AR_f8.any(out=B)) # E: SubClass reveal_type(f8.argmax()) # E: {intp} reveal_type(AR_f8.argmax()) # E: {intp} reveal_type(AR_f8.argmax(axis=0)) # E: Any reveal_type(AR_f8.argmax(out=B)) # E: SubClass reveal_type(f8.argmin()) # E: {intp} reveal_type(AR_f8.argmin()) # E: {intp} reveal_type(AR_f8.argmin(axis=0)) # E: Any reveal_type(AR_f8.argmin(out=B)) # E: SubClass reveal_type(f8.argsort()) # E: ndarray[Any, Any] reveal_type(AR_f8.argsort()) # E: ndarray[Any, Any] reveal_type(f8.astype(np.int64).choose([()])) # E: ndarray[Any, Any] reveal_type(AR_f8.choose([0])) # E: ndarray[Any, Any] reveal_type(AR_f8.choose([0], out=B)) # E: SubClass reveal_type(f8.clip(1)) # E: Any reveal_type(AR_f8.clip(1)) # E: Any reveal_type(AR_f8.clip(None, 1)) # E: Any reveal_type(AR_f8.clip(1, out=B)) # E: SubClass reveal_type(AR_f8.clip(None, 1, out=B)) # E: SubClass reveal_type(f8.compress([0])) # E: ndarray[Any, Any] reveal_type(AR_f8.compress([0])) # E: ndarray[Any, Any] reveal_type(AR_f8.compress([0], out=B)) # E: SubClass reveal_type(f8.conj()) # E: {float64} reveal_type(AR_f8.conj()) # E: ndarray[Any, dtype[{float64}]] reveal_type(B.conj()) # E: SubClass reveal_type(f8.conjugate()) # E: {float64} reveal_type(AR_f8.conjugate()) # E: ndarray[Any, dtype[{float64}]] reveal_type(B.conjugate()) # E: SubClass reveal_type(f8.cumprod()) # E: ndarray[Any, Any] reveal_type(AR_f8.cumprod()) # E: ndarray[Any, Any] reveal_type(AR_f8.cumprod(out=B)) # E: SubClass reveal_type(f8.cumsum()) # E: ndarray[Any, Any] reveal_type(AR_f8.cumsum()) # E: ndarray[Any, Any] reveal_type(AR_f8.cumsum(out=B)) # E: SubClass reveal_type(f8.max()) # E: Any reveal_type(AR_f8.max()) # E: Any reveal_type(AR_f8.max(axis=0)) # E: Any reveal_type(AR_f8.max(keepdims=True)) # E: Any reveal_type(AR_f8.max(out=B)) # E: SubClass reveal_type(f8.mean()) # E: Any reveal_type(AR_f8.mean()) # E: Any reveal_type(AR_f8.mean(axis=0)) # E: Any reveal_type(AR_f8.mean(keepdims=True)) # E: Any reveal_type(AR_f8.mean(out=B)) # E: SubClass reveal_type(f8.min()) # E: Any reveal_type(AR_f8.min()) # E: Any reveal_type(AR_f8.min(axis=0)) # E: Any reveal_type(AR_f8.min(keepdims=True)) # E: Any reveal_type(AR_f8.min(out=B)) # E: SubClass reveal_type(f8.newbyteorder()) # E: {float64} reveal_type(AR_f8.newbyteorder()) # E: ndarray[Any, dtype[{float64}]] reveal_type(B.newbyteorder('|')) # E: SubClass reveal_type(f8.prod()) # E: Any reveal_type(AR_f8.prod()) # E: Any reveal_type(AR_f8.prod(axis=0)) # E: Any reveal_type(AR_f8.prod(keepdims=True)) # E: Any reveal_type(AR_f8.prod(out=B)) # E: SubClass reveal_type(f8.ptp()) # E: Any reveal_type(AR_f8.ptp()) # E: Any reveal_type(AR_f8.ptp(axis=0)) # E: Any reveal_type(AR_f8.ptp(keepdims=True)) # E: Any reveal_type(AR_f8.ptp(out=B)) # E: SubClass reveal_type(f8.round()) # E: {float64} reveal_type(AR_f8.round()) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_f8.round(out=B)) # E: SubClass reveal_type(f8.repeat(1)) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_f8.repeat(1)) # E: ndarray[Any, dtype[{float64}]] reveal_type(B.repeat(1)) # E: ndarray[Any, dtype[object_]] reveal_type(f8.std()) # E: Any reveal_type(AR_f8.std()) # E: Any reveal_type(AR_f8.std(axis=0)) # E: Any reveal_type(AR_f8.std(keepdims=True)) # E: Any reveal_type(AR_f8.std(out=B)) # E: SubClass reveal_type(f8.sum()) # E: Any reveal_type(AR_f8.sum()) # E: Any reveal_type(AR_f8.sum(axis=0)) # E: Any reveal_type(AR_f8.sum(keepdims=True)) # E: Any reveal_type(AR_f8.sum(out=B)) # E: SubClass reveal_type(f8.take(0)) # E: {float64} reveal_type(AR_f8.take(0)) # E: {float64} reveal_type(AR_f8.take([0])) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_f8.take(0, out=B)) # E: SubClass reveal_type(AR_f8.take([0], out=B)) # E: SubClass reveal_type(f8.var()) # E: Any reveal_type(AR_f8.var()) # E: Any reveal_type(AR_f8.var(axis=0)) # E: Any reveal_type(AR_f8.var(keepdims=True)) # E: Any reveal_type(AR_f8.var(out=B)) # E: SubClass reveal_type(AR_f8.argpartition([0])) # E: ndarray[Any, dtype[{intp}]] reveal_type(AR_f8.diagonal()) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_f8.dot(1)) # E: ndarray[Any, Any] reveal_type(AR_f8.dot([1])) # E: Any reveal_type(AR_f8.dot(1, out=B)) # E: SubClass reveal_type(AR_f8.nonzero()) # E: tuple[ndarray[Any, dtype[{intp}]], ...] reveal_type(AR_f8.searchsorted(1)) # E: {intp} reveal_type(AR_f8.searchsorted([1])) # E: ndarray[Any, dtype[{intp}]] reveal_type(AR_f8.trace()) # E: Any reveal_type(AR_f8.trace(out=B)) # E: SubClass reveal_type(AR_f8.item()) # E: float reveal_type(AR_U.item()) # E: str reveal_type(AR_f8.ravel()) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_U.ravel()) # E: ndarray[Any, dtype[str_]] reveal_type(AR_f8.flatten()) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_U.flatten()) # E: ndarray[Any, dtype[str_]] reveal_type(AR_f8.reshape(1)) # E: ndarray[Any, dtype[{float64}]] reveal_type(AR_U.reshape(1)) # E: ndarray[Any, dtype[str_]] reveal_type(int(AR_f8)) # E: int reveal_type(int(AR_U)) # E: int reveal_type(float(AR_f8)) # E: float reveal_type(float(AR_U)) # E: float reveal_type(complex(AR_f8)) # E: complex reveal_type(operator.index(AR_i8)) # E: int reveal_type(AR_f8.__array_prepare__(B)) # E: ndarray[Any, dtype[object_]] reveal_type(AR_f8.__array_wrap__(B)) # E: ndarray[Any, dtype[object_]] reveal_type(AR_V[0]) # E: Any reveal_type(AR_V[0, 0]) # E: Any reveal_type(AR_V[AR_i8]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[AR_i8, AR_i8]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[AR_i8, None]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[0, ...]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[[0]]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[[0], [0]]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[:]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V["a"]) # E: ndarray[Any, dtype[Any]] reveal_type(AR_V[["a", "b"]]) # E: ndarray[Any, dtype[void]] reveal_type(AR_f8.dump("test_file")) # E: None reveal_type(AR_f8.dump(b"test_file")) # E: None with open("test_file", "wb") as f: reveal_type(AR_f8.dump(f)) # E: None reveal_type(AR_f8.__array_finalize__(None)) # E: None reveal_type(AR_f8.__array_finalize__(B)) # E: None reveal_type(AR_f8.__array_finalize__(AR_f8)) # E: None