Hacked By AnonymousFox

"""
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