Skip to content

Commit 6a4f832

Browse files
committed
Update tests forn NEP-50
1 parent 27862ea commit 6a4f832

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

dpnp/tests/third_party/cupy/core_tests/test_nep50_examples.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import pytest
33

44
import dpnp as cp
5+
from dpnp.tests.helper import has_support_aspect64
56
from dpnp.tests.third_party.cupy import testing
67

78
# TODO: remove once all dtype aliases added
@@ -47,20 +48,23 @@
4748
@pytest.mark.filterwarnings("ignore::RuntimeWarning")
4849
@testing.with_requires("numpy>=2.0")
4950
@pytest.mark.parametrize("example", examples)
50-
@testing.numpy_cupy_allclose(atol=1e-15, accept_error=OverflowError)
51+
@testing.numpy_cupy_allclose(
52+
atol=1e-15, accept_error=OverflowError, type_check=has_support_aspect64()
53+
)
5154
def test_nep50_examples(xp, example):
5255
dct = {
5356
"array": xp.array,
5457
"uint8": xp.uint8,
5558
"int64": xp.int64,
5659
"float32": xp.float32,
57-
"float64": xp.float64,
5860
"int16": xp.int16,
5961
"bool_": xp.bool_,
6062
"int32": xp.int32,
6163
"complex64": xp.complex64,
6264
"int8": xp.int8,
6365
}
66+
if has_support_aspect64():
67+
dct["float64"] = xp.float64
6468

6569
if isinstance(example, tuple):
6670
example, mesg = example

0 commit comments

Comments
 (0)