@@ -162,6 +162,8 @@ def test_convolve(self, a, v, mode, dtype, method):
162
162
@pytest .mark .parametrize ("dtype" , get_all_dtypes (no_none = True ))
163
163
@pytest .mark .parametrize ("method" , ["auto" , "direct" , "fft" ])
164
164
def test_convolve_random (self , a_size , v_size , mode , dtype , method ):
165
+ if dtype in [numpy .int8 , numpy .uint8 , numpy .int16 , numpy .uint16 ]:
166
+ pytest .skip ("avoid overflow." )
165
167
if dtype == dpnp .bool :
166
168
an = numpy .random .rand (a_size ) > 0.9
167
169
vn = numpy .random .rand (v_size ) > 0.9
@@ -385,7 +387,7 @@ def test_correlate(self, a, v, mode, dtype, method):
385
387
@pytest .mark .parametrize ("dtype" , get_all_dtypes (no_none = True ))
386
388
@pytest .mark .parametrize ("method" , ["auto" , "direct" , "fft" ])
387
389
def test_correlate_random (self , a_size , v_size , mode , dtype , method ):
388
- if dtype in [numpy .int8 , numpy .uint8 ]:
390
+ if dtype in [numpy .int8 , numpy .uint8 , numpy . int16 , numpy . uint16 ]:
389
391
pytest .skip ("avoid overflow." )
390
392
an = generate_random_numpy_array (
391
393
a_size , dtype , low = - 3 , high = 3 , probability = 0.9
0 commit comments