1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
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
|
