@@ -166,6 +166,7 @@ def test_extract_no_bool(self, xp, dtype):
166
166
b = xp .array ([[1 , 0 , 1 ], [0 , 1 , 0 ], [1 , 0 , 1 ]], dtype = dtype )
167
167
return xp .extract (b , a )
168
168
169
+ @pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
169
170
@testing .numpy_cupy_array_equal ()
170
171
def test_extract_shape_mismatch (self , xp ):
171
172
a = testing .shaped_arange ((2 , 3 ), xp )
@@ -174,20 +175,23 @@ def test_extract_shape_mismatch(self, xp):
174
175
[True , False ]])
175
176
return xp .extract (b , a )
176
177
178
+ @pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
177
179
@testing .numpy_cupy_array_equal ()
178
180
def test_extract_size_mismatch (self , xp ):
179
181
a = testing .shaped_arange ((3 , 3 ), xp )
180
182
b = xp .array ([[True , False , True ],
181
183
[False , True , False ]])
182
184
return xp .extract (b , a )
183
185
186
+ @pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
184
187
@testing .numpy_cupy_array_equal ()
185
188
def test_extract_size_mismatch2 (self , xp ):
186
189
a = testing .shaped_arange ((3 , 3 ), xp )
187
190
b = xp .array ([[True , False , True , False ],
188
191
[False , True , False , True ]])
189
192
return xp .extract (b , a )
190
193
194
+ @pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
191
195
@testing .numpy_cupy_array_equal ()
192
196
def test_extract_empty_1dim (self , xp ):
193
197
a = testing .shaped_arange ((3 , 3 ), xp )
0 commit comments