52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
import numpy as np
|
|
import numpy.typing as npt
|
|
|
|
nd: npt.NDArray[np.int_] = np.array([[1, 2], [3, 4]])
|
|
|
|
# item
|
|
reveal_type(nd.item()) # E: int
|
|
reveal_type(nd.item(1)) # E: int
|
|
reveal_type(nd.item(0, 1)) # E: int
|
|
reveal_type(nd.item((0, 1))) # E: int
|
|
|
|
# tolist
|
|
reveal_type(nd.tolist()) # E: Any
|
|
|
|
# itemset does not return a value
|
|
# tostring is pretty simple
|
|
# tobytes is pretty simple
|
|
# tofile does not return a value
|
|
# dump does not return a value
|
|
# dumps is pretty simple
|
|
|
|
# astype
|
|
reveal_type(nd.astype("float")) # E: ndarray[Any, dtype[Any]]
|
|
reveal_type(nd.astype(float)) # E: ndarray[Any, dtype[Any]]
|
|
reveal_type(nd.astype(np.float64)) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(nd.astype(np.float64, "K")) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(nd.astype(np.float64, "K", "unsafe")) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(nd.astype(np.float64, "K", "unsafe", True)) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(nd.astype(np.float64, "K", "unsafe", True, True)) # E: ndarray[Any, dtype[{float64}]]
|
|
|
|
# byteswap
|
|
reveal_type(nd.byteswap()) # E: ndarray[Any, dtype[{int_}]]
|
|
reveal_type(nd.byteswap(True)) # E: ndarray[Any, dtype[{int_}]]
|
|
|
|
# copy
|
|
reveal_type(nd.copy()) # E: ndarray[Any, dtype[{int_}]]
|
|
reveal_type(nd.copy("C")) # E: ndarray[Any, dtype[{int_}]]
|
|
|
|
reveal_type(nd.view()) # E: ndarray[Any, dtype[{int_}]]
|
|
reveal_type(nd.view(np.float64)) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(nd.view(float)) # E: ndarray[Any, dtype[Any]]
|
|
reveal_type(nd.view(np.float64, np.matrix)) # E: matrix[Any, Any]
|
|
|
|
# getfield
|
|
reveal_type(nd.getfield("float")) # E: ndarray[Any, dtype[Any]]
|
|
reveal_type(nd.getfield(float)) # E: ndarray[Any, dtype[Any]]
|
|
reveal_type(nd.getfield(np.float64)) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(nd.getfield(np.float64, 8)) # E: ndarray[Any, dtype[{float64}]]
|
|
|
|
# setflags does not return a value
|
|
# fill does not return a value
|