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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
"""
Tests for :mod:`_core.numeric`.
Does not include tests which fall under ``array_constructors``.
"""
from typing import Any, assert_type
import numpy as np
import numpy.typing as npt
class SubClass(npt.NDArray[np.int64]): ...
i8: np.int64
AR_b: npt.NDArray[np.bool]
AR_u8: npt.NDArray[np.uint64]
AR_i8: npt.NDArray[np.int64]
AR_f8: npt.NDArray[np.float64]
AR_c16: npt.NDArray[np.complex128]
AR_m: npt.NDArray[np.timedelta64]
AR_O: npt.NDArray[np.object_]
B: list[int]
C: SubClass
assert_type(np.count_nonzero(i8), np.intp)
assert_type(np.count_nonzero(AR_i8), np.intp)
assert_type(np.count_nonzero(B), np.intp)
assert_type(np.count_nonzero(AR_i8, keepdims=True), npt.NDArray[np.intp])
assert_type(np.count_nonzero(AR_i8, axis=0), Any)
assert_type(np.isfortran(i8), bool)
assert_type(np.isfortran(AR_i8), bool)
assert_type(np.argwhere(i8), npt.NDArray[np.intp])
assert_type(np.argwhere(AR_i8), npt.NDArray[np.intp])
assert_type(np.flatnonzero(i8), npt.NDArray[np.intp])
assert_type(np.flatnonzero(AR_i8), npt.NDArray[np.intp])
assert_type(np.correlate(B, AR_i8, mode="valid"), npt.NDArray[np.signedinteger])
assert_type(np.correlate(AR_i8, AR_i8, mode="same"), npt.NDArray[np.signedinteger])
assert_type(np.correlate(AR_b, AR_b), npt.NDArray[np.bool])
assert_type(np.correlate(AR_b, AR_u8), npt.NDArray[np.unsignedinteger])
assert_type(np.correlate(AR_i8, AR_b), npt.NDArray[np.signedinteger])
assert_type(np.correlate(AR_i8, AR_f8), npt.NDArray[np.floating])
assert_type(np.correlate(AR_i8, AR_c16), npt.NDArray[np.complexfloating])
assert_type(np.correlate(AR_i8, AR_m), npt.NDArray[np.timedelta64])
assert_type(np.correlate(AR_O, AR_O), npt.NDArray[np.object_])
assert_type(np.convolve(B, AR_i8, mode="valid"), npt.NDArray[np.signedinteger])
assert_type(np.convolve(AR_i8, AR_i8, mode="same"), npt.NDArray[np.signedinteger])
assert_type(np.convolve(AR_b, AR_b), npt.NDArray[np.bool])
assert_type(np.convolve(AR_b, AR_u8), npt.NDArray[np.unsignedinteger])
assert_type(np.convolve(AR_i8, AR_b), npt.NDArray[np.signedinteger])
assert_type(np.convolve(AR_i8, AR_f8), npt.NDArray[np.floating])
assert_type(np.convolve(AR_i8, AR_c16), npt.NDArray[np.complexfloating])
assert_type(np.convolve(AR_i8, AR_m), npt.NDArray[np.timedelta64])
assert_type(np.convolve(AR_O, AR_O), npt.NDArray[np.object_])
assert_type(np.outer(i8, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.outer(B, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.outer(AR_i8, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.outer(AR_i8, AR_i8, out=C), SubClass)
assert_type(np.outer(AR_b, AR_b), npt.NDArray[np.bool])
assert_type(np.outer(AR_b, AR_u8), npt.NDArray[np.unsignedinteger])
assert_type(np.outer(AR_i8, AR_b), npt.NDArray[np.signedinteger])
assert_type(np.convolve(AR_i8, AR_f8), npt.NDArray[np.floating])
assert_type(np.outer(AR_i8, AR_c16), npt.NDArray[np.complexfloating])
assert_type(np.outer(AR_i8, AR_m), npt.NDArray[np.timedelta64])
assert_type(np.outer(AR_O, AR_O), npt.NDArray[np.object_])
assert_type(np.tensordot(B, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.tensordot(AR_i8, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.tensordot(AR_i8, AR_i8, axes=0), npt.NDArray[np.signedinteger])
assert_type(np.tensordot(AR_i8, AR_i8, axes=(0, 1)), npt.NDArray[np.signedinteger])
assert_type(np.tensordot(AR_b, AR_b), npt.NDArray[np.bool])
assert_type(np.tensordot(AR_b, AR_u8), npt.NDArray[np.unsignedinteger])
assert_type(np.tensordot(AR_i8, AR_b), npt.NDArray[np.signedinteger])
assert_type(np.tensordot(AR_i8, AR_f8), npt.NDArray[np.floating])
assert_type(np.tensordot(AR_i8, AR_c16), npt.NDArray[np.complexfloating])
assert_type(np.tensordot(AR_i8, AR_m), npt.NDArray[np.timedelta64])
assert_type(np.tensordot(AR_O, AR_O), npt.NDArray[np.object_])
assert_type(np.isscalar(i8), bool)
assert_type(np.isscalar(AR_i8), bool)
assert_type(np.isscalar(B), bool)
assert_type(np.roll(AR_i8, 1), npt.NDArray[np.int64])
assert_type(np.roll(AR_i8, (1, 2)), npt.NDArray[np.int64])
assert_type(np.roll(B, 1), npt.NDArray[Any])
assert_type(np.rollaxis(AR_i8, 0, 1), npt.NDArray[np.int64])
assert_type(np.moveaxis(AR_i8, 0, 1), npt.NDArray[np.int64])
assert_type(np.moveaxis(AR_i8, (0, 1), (1, 2)), npt.NDArray[np.int64])
assert_type(np.cross(B, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.cross(AR_i8, AR_i8), npt.NDArray[np.signedinteger])
assert_type(np.cross(AR_b, AR_u8), npt.NDArray[np.unsignedinteger])
assert_type(np.cross(AR_i8, AR_b), npt.NDArray[np.signedinteger])
assert_type(np.cross(AR_i8, AR_f8), npt.NDArray[np.floating])
assert_type(np.cross(AR_i8, AR_c16), npt.NDArray[np.complexfloating])
assert_type(np.cross(AR_O, AR_O), npt.NDArray[np.object_])
assert_type(np.indices([0, 1, 2]), npt.NDArray[np.int_])
assert_type(np.indices([0, 1, 2], sparse=True), tuple[npt.NDArray[np.int_], ...])
assert_type(np.indices([0, 1, 2], dtype=np.float64), npt.NDArray[np.float64])
assert_type(np.indices([0, 1, 2], sparse=True, dtype=np.float64), tuple[npt.NDArray[np.float64], ...])
assert_type(np.indices([0, 1, 2], dtype=float), npt.NDArray[Any])
assert_type(np.indices([0, 1, 2], sparse=True, dtype=float), tuple[npt.NDArray[Any], ...])
assert_type(np.binary_repr(1), str)
assert_type(np.base_repr(1), str)
assert_type(np.allclose(i8, AR_i8), bool)
assert_type(np.allclose(B, AR_i8), bool)
assert_type(np.allclose(AR_i8, AR_i8), bool)
assert_type(np.isclose(i8, i8), np.bool)
assert_type(np.isclose(i8, AR_i8), npt.NDArray[np.bool])
assert_type(np.isclose(B, AR_i8), npt.NDArray[np.bool])
assert_type(np.isclose(AR_i8, AR_i8), npt.NDArray[np.bool])
assert_type(np.array_equal(i8, AR_i8), bool)
assert_type(np.array_equal(B, AR_i8), bool)
assert_type(np.array_equal(AR_i8, AR_i8), bool)
assert_type(np.array_equiv(i8, AR_i8), bool)
assert_type(np.array_equiv(B, AR_i8), bool)
assert_type(np.array_equiv(AR_i8, AR_i8), bool)
|
