Skip to content

Commit ec65f73

Browse files
Skip [u]int8/16 tests for convolve due to overflow
1 parent 886b0b8 commit ec65f73

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpnp/tests/test_statistics.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,8 @@ def test_convolve(self, a, v, mode, dtype, method):
162162
@pytest.mark.parametrize("dtype", get_all_dtypes(no_none=True))
163163
@pytest.mark.parametrize("method", ["auto", "direct", "fft"])
164164
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.")
165167
if dtype == dpnp.bool:
166168
an = numpy.random.rand(a_size) > 0.9
167169
vn = numpy.random.rand(v_size) > 0.9
@@ -385,7 +387,7 @@ def test_correlate(self, a, v, mode, dtype, method):
385387
@pytest.mark.parametrize("dtype", get_all_dtypes(no_none=True))
386388
@pytest.mark.parametrize("method", ["auto", "direct", "fft"])
387389
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]:
389391
pytest.skip("avoid overflow.")
390392
an = generate_random_numpy_array(
391393
a_size, dtype, low=-3, high=3, probability=0.9

0 commit comments

Comments
 (0)