|
42 | 42 | NUMPY_REVEALED_TYPES = [
|
43 | 43 | (
|
44 | 44 | 'arrays(dtype=np.dtype("int32"), shape=1)',
|
45 |
| - "ndarray[Any, dtype[signedinteger[_32Bit]]]", |
| 45 | + "ndarray[tuple[int, ...], dtype[signedinteger[_32Bit]]]", |
46 | 46 | ),
|
47 | 47 | (
|
48 | 48 | "arrays(dtype=np.dtype(int), shape=1)",
|
49 |
| - "ndarray[Any, dtype[signedinteger[Any]]]", |
| 49 | + "ndarray[tuple[int, ...], dtype[Union[signedinteger[Union[_32Bit, _64Bit]], bool[bool]]]]", |
50 | 50 | ),
|
51 | 51 | (
|
52 | 52 | "boolean_dtypes()",
|
53 |
| - "dtype[bool_]" if NP1 else "dtype[bool]", |
| 53 | + "dtype[bool[bool]]", # np.bool[builtins.bool] |
54 | 54 | ),
|
55 | 55 | (
|
56 | 56 | "unsigned_integer_dtypes(sizes=8)",
|
|
110 | 110 | ),
|
111 | 111 | (
|
112 | 112 | "floating_dtypes(sizes=64)",
|
113 |
| - "dtype[floating[_64Bit]]", |
| 113 | + "dtype[float64]", |
114 | 114 | ),
|
115 | 115 | (
|
116 | 116 | "floating_dtypes(sizes=128)",
|
|
130 | 130 | ),
|
131 | 131 | (
|
132 | 132 | "complex_number_dtypes(sizes=128)",
|
133 |
| - "dtype[complexfloating[_64Bit, _64Bit]]", |
| 133 | + "dtype[complex128]", |
134 | 134 | ),
|
135 | 135 | (
|
136 | 136 | "complex_number_dtypes(sizes=256)",
|
|
146 | 146 | ),
|
147 | 147 | (
|
148 | 148 | "integer_array_indices(shape=(2, 3))",
|
149 |
| - "tuple[ndarray[Any, dtype[signedinteger[Any]]], ...]", |
| 149 | + "tuple[ndarray[tuple[int, ...], dtype[signedinteger[Any]]], ...]", |
150 | 150 | ),
|
151 | 151 | (
|
152 | 152 | 'integer_array_indices(shape=(2, 3), dtype=np.dtype("int32"))',
|
153 |
| - "tuple[ndarray[Any, dtype[signedinteger[_32Bit]]], ...]", |
| 153 | + "tuple[ndarray[tuple[int, ...], dtype[signedinteger[_32Bit]]], ...]", |
154 | 154 | ),
|
155 | 155 | (
|
156 | 156 | 'integer_array_indices(shape=(2, 3), dtype=np.dtype("uint8"))',
|
157 |
| - "tuple[ndarray[Any, dtype[unsignedinteger[_8Bit]]], ...]", |
| 157 | + "tuple[ndarray[tuple[int, ...], dtype[unsignedinteger[_8Bit]]], ...]", |
158 | 158 | ),
|
159 | 159 | ]
|
0 commit comments