codekingpro/portable-devtools

1"""Tests for :mod:`numpy._core.fromnumeric`."""
2
3import numpy as np
4import numpy.typing as npt
5
6A = np.array(True, ndmin=2, dtype=bool)
7A.setflags(write=False)
8AR_U: npt.NDArray[np.str_]
9AR_M: npt.NDArray[np.datetime64]
10AR_f4: npt.NDArray[np.float32]
11
12a = np.bool(True)
13
14np.take(a, None)  # type: ignore[call-overload]
15np.take(a, axis=1.0)  # type: ignore[call-overload]
16np.take(A, out=1)  # type: ignore[call-overload]
17np.take(A, mode="bob")  # type: ignore[call-overload]
18
19np.reshape(a, None)  # type: ignore[call-overload]
20np.reshape(A, 1, order="bob")  # type: ignore[call-overload]
21
22np.choose(a, None)  # type: ignore[call-overload]
23np.choose(a, out=1.0)  # type: ignore[call-overload]
24np.choose(A, mode="bob")  # type: ignore[call-overload]
25
26np.repeat(a, None)  # type: ignore[call-overload]
27np.repeat(A, 1, axis=1.0)  # type: ignore[call-overload]
28
29np.swapaxes(A, None, 1)  # type: ignore[call-overload]
30np.swapaxes(A, 1, [0])  # type: ignore[call-overload]
31
32np.transpose(A, axes=1.0)  # type: ignore[call-overload]
33
34np.partition(a, None)  # type: ignore[call-overload]
35np.partition(a, 0, axis="bob")  # type: ignore[call-overload]
36np.partition(A, 0, kind="bob")  # type: ignore[call-overload]
37np.partition(A, 0, order=range(5))  # type: ignore[arg-type]
38
39np.argpartition(a, None)  # type: ignore[arg-type]
40np.argpartition(a, 0, axis="bob")  # type: ignore[arg-type]
41np.argpartition(A, 0, kind="bob")  # type: ignore[arg-type]
42np.argpartition(A, 0, order=range(5))  # type: ignore[arg-type]
43
44np.sort(A, axis="bob")  # type: ignore[call-overload]
45np.sort(A, kind="bob")  # type: ignore[call-overload]
46np.sort(A, order=range(5))  # type: ignore[arg-type]
47
48np.argsort(A, axis="bob")  # type: ignore[arg-type]
49np.argsort(A, kind="bob")  # type: ignore[arg-type]
50np.argsort(A, order=range(5))  # type: ignore[arg-type]
51
52np.argmax(A, axis="bob")  # type: ignore[call-overload]
53np.argmax(A, kind="bob")  # type: ignore[call-overload]
54np.argmax(A, out=AR_f4)  # type: ignore[type-var]
55
56np.argmin(A, axis="bob")  # type: ignore[call-overload]
57np.argmin(A, kind="bob")  # type: ignore[call-overload]
58np.argmin(A, out=AR_f4)  # type: ignore[type-var]
59
60np.searchsorted(A[0], 0, side="bob")  # type: ignore[call-overload]
61np.searchsorted(A[0], 0, sorter=1.0)  # type: ignore[call-overload]
62
63np.resize(A, 1.0)  # type: ignore[call-overload]
64
65np.squeeze(A, 1.0)  # type: ignore[call-overload]
66
67np.diagonal(A, offset=None)  # type: ignore[call-overload]
68np.diagonal(A, axis1="bob")  # type: ignore[call-overload]
69np.diagonal(A, axis2=[])  # type: ignore[call-overload]
70
71np.trace(A, offset=None)  # type: ignore[call-overload]
72np.trace(A, axis1="bob")  # type: ignore[call-overload]
73np.trace(A, axis2=[])  # type: ignore[call-overload]
74
75np.ravel(a, order="bob")  # type: ignore[call-overload]
76
77np.nonzero(0)  # type: ignore[arg-type]
78
79np.compress([True], A, axis=1.0)  # type: ignore[call-overload]
80
81np.clip(a, 1, 2, out=1)  # type: ignore[call-overload]
82
83np.sum(a, axis=1.0)  # type: ignore[call-overload]
84np.sum(a, keepdims=1.0)  # type: ignore[call-overload]
85np.sum(a, initial=[1])  # type: ignore[call-overload]
86
87np.all(a, axis=1.0)  # type: ignore[call-overload]
88np.all(a, keepdims=1.0)  # type: ignore[call-overload]
89np.all(a, out=1.0)  # type: ignore[call-overload]
90
91np.any(a, axis=1.0)  # type: ignore[call-overload]
92np.any(a, keepdims=1.0)  # type: ignore[call-overload]
93np.any(a, out=1.0)  # type: ignore[call-overload]
94
95np.cumsum(a, axis=1.0)  # type: ignore[call-overload]
96np.cumsum(a, dtype=1.0)  # type: ignore[call-overload]
97np.cumsum(a, out=1.0)  # type: ignore[call-overload]
98
99np.ptp(a, axis=1.0)  # type: ignore[call-overload]
100np.ptp(a, keepdims=1.0)  # type: ignore[call-overload]
101np.ptp(a, out=1.0)  # type: ignore[call-overload]
102
103np.amax(a, axis=1.0)  # type: ignore[call-overload]
104np.amax(a, keepdims=1.0)  # type: ignore[call-overload]
105np.amax(a, out=1.0)  # type: ignore[call-overload]
106np.amax(a, initial=[1.0])  # type: ignore[call-overload]
107np.amax(a, where=[1.0])  # type: ignore[arg-type]
108
109np.amin(a, axis=1.0)  # type: ignore[call-overload]
110np.amin(a, keepdims=1.0)  # type: ignore[call-overload]
111np.amin(a, out=1.0)  # type: ignore[call-overload]
112np.amin(a, initial=[1.0])  # type: ignore[call-overload]
113np.amin(a, where=[1.0])  # type: ignore[arg-type]
114
115np.prod(a, axis=1.0)  # type: ignore[call-overload]
116np.prod(a, out=False)  # type: ignore[call-overload]
117np.prod(a, keepdims=1.0)  # type: ignore[call-overload]
118np.prod(a, initial=int)  # type: ignore[call-overload]
119np.prod(a, where=1.0)  # type: ignore[call-overload]
120np.prod(AR_U)  # type: ignore[arg-type]
121
122np.cumprod(a, axis=1.0)  # type: ignore[call-overload]
123np.cumprod(a, out=False)  # type: ignore[call-overload]
124np.cumprod(AR_U)  # type: ignore[arg-type]
125
126np.size(a, axis=1.0)  # type: ignore[arg-type]
127
128np.around(a, decimals=1.0)  # type: ignore[call-overload]
129np.around(a, out=type)  # type: ignore[call-overload]
130np.around(AR_U)  # type: ignore[arg-type]
131
132np.mean(a, axis=1.0)  # type: ignore[call-overload]
133np.mean(a, out=False)  # type: ignore[call-overload]
134np.mean(a, keepdims=1.0)  # type: ignore[call-overload]
135np.mean(AR_U)  # type: ignore[arg-type]
136np.mean(AR_M)  # type: ignore[arg-type]
137
138np.std(a, axis=1.0)  # type: ignore[call-overload]
139np.std(a, out=False)  # type: ignore[call-overload]
140np.std(a, ddof="test")  # type: ignore[call-overload]
141np.std(a, keepdims=1.0)  # type: ignore[call-overload]
142np.std(AR_U)  # type: ignore[arg-type]
143
144np.var(a, axis=1.0)  # type: ignore[call-overload]
145np.var(a, out=False)  # type: ignore[call-overload]
146np.var(a, ddof="test")  # type: ignore[call-overload]
147np.var(a, keepdims=1.0)  # type: ignore[call-overload]
148np.var(AR_U)  # type: ignore[arg-type]
149