Skip to content

Commit e7569d0

Browse files
author
Vahid Tavanashad
committed
update test to pass with numpy-2.3
1 parent 2d27110 commit e7569d0

File tree

6 files changed

+15
-10
lines changed

6 files changed

+15
-10
lines changed

.github/workflows/conda-package.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ env:
2222
test-env-name: 'test'
2323
rerun-tests-on-failure: 'true'
2424
rerun-tests-max-attempts: 2
25-
rerun-tests-timeout: 35
25+
rerun-tests-timeout: 40
2626

2727
jobs:
2828
build:

dpnp/tests/test_linalg.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2104,11 +2104,9 @@ def test_empty(self, shape, ord, axis, keepdims):
21042104
assert_raises(ValueError, dpnp.linalg.norm, ia, **kwarg)
21052105
assert_raises(ValueError, numpy.linalg.norm, a, **kwarg)
21062106
else:
2107-
# TODO: when similar changes in numpy are available, instead
2108-
# of assert_equal with zero, we should compare with numpy
2109-
# ord in [None, 1, 2]
2110-
assert_equal(dpnp.linalg.norm(ia, **kwarg), 0.0)
2111-
assert_raises(ValueError, numpy.linalg.norm, a, **kwarg)
2107+
result = dpnp.linalg.norm(ia, **kwarg)
2108+
expected = numpy.linalg.norm(a, **kwarg)
2109+
assert_dtype_allclose(result, expected)
21122110
else:
21132111
result = dpnp.linalg.norm(ia, **kwarg)
21142112
expected = numpy.linalg.norm(a, **kwarg)

dpnp/tests/test_product.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,6 @@
1212
assert_dtype_allclose,
1313
generate_random_numpy_array,
1414
get_all_dtypes,
15-
get_complex_dtypes,
16-
is_win_platform,
1715
numpy_version,
1816
)
1917
from .third_party.cupy import testing
@@ -845,6 +843,8 @@ def test_dtype_matrix(self, dt_in1, dt_in2, dt_out, shape1, shape2):
845843
assert_raises(TypeError, dpnp.matmul, ia, ib, out=iout)
846844
assert_raises(TypeError, numpy.matmul, a, b, out=out)
847845

846+
# TODO: include numpy-2.3 when numpy-issue-29164 is resolved
847+
@testing.with_requires("numpy<2.3")
848848
@pytest.mark.parametrize("dtype", _selected_dtypes)
849849
@pytest.mark.parametrize("order1", ["C", "F", "A"])
850850
@pytest.mark.parametrize("order2", ["C", "F", "A"])
@@ -882,6 +882,8 @@ def test_order(self, dtype, order1, order2, order, shape1, shape2):
882882
assert result.flags.f_contiguous == expected.flags.f_contiguous
883883
assert_dtype_allclose(result, expected)
884884

885+
# TODO: include numpy-2.3 when numpy-issue-29164 is resolved
886+
@testing.with_requires("numpy<2.3")
885887
@pytest.mark.parametrize("dtype", _selected_dtypes)
886888
@pytest.mark.parametrize(
887889
"stride",

dpnp/tests/testing/array.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@
2929

3030
from dpnp.dpnp_utils import convert_item
3131

32+
from ..helper import numpy_version
33+
3234
assert_allclose_orig = numpy.testing.assert_allclose
3335
assert_almost_equal_orig = numpy.testing.assert_almost_equal
3436
assert_array_almost_equal_orig = numpy.testing.assert_array_almost_equal
@@ -49,7 +51,7 @@ def _assert(assert_func, result, expected, *args, **kwargs):
4951
]
5052
# For numpy < 2.0, some tests will fail for dtype mismatch
5153
dev = dpctl.select_default_device()
52-
if numpy.__version__ >= "2.0.0" and dev.has_aspect_fp64:
54+
if numpy_version() >= "2.0.0" and dev.has_aspect_fp64:
5355
strict = kwargs.setdefault("strict", True)
5456
if flag:
5557
if strict:

dpnp/tests/third_party/cupy/manipulation_tests/test_add_remove.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,8 @@ def test_unique_inverse(self, xp, dtype, attr):
340340
a = testing.shaped_random((100, 100), xp, dtype)
341341
return getattr(xp.unique_inverse(a), attr)
342342

343-
@testing.with_requires("numpy>=2.0")
343+
# TODO: include numpy-2.3 when dpnp-issue-2476 is addressed
344+
@testing.with_requires("numpy>=2.0", "numpy<2.3")
344345
@testing.for_all_dtypes(no_float16=True, no_bool=True, no_complex=True)
345346
@testing.numpy_cupy_array_equal()
346347
def test_unique_values(self, xp, dtype):

dpnp/tests/third_party/cupy/math_tests/test_matmul.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,8 @@ def test_cupy_matmul(self, xp, dtype1, dtype2):
8383
return xp.matmul(x1, x2)
8484

8585

86+
# TODO: include numpy-2.3 when numpy-issue-29164 is resolved
87+
@testing.with_requires("numpy<2.3")
8688
@testing.parameterize(
8789
*testing.product(
8890
{

0 commit comments

Comments
 (0)