Skip to content

Commit 2da72cd

Browse files
committed
Update sorting_tests/test_count.py
1 parent 759e916 commit 2da72cd

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

dpnp/tests/third_party/cupy/sorting_tests/test_count.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88

99
class TestCount(unittest.TestCase):
10+
1011
@testing.for_all_dtypes()
1112
def test_count_nonzero(self, dtype):
1213
def func(xp):
@@ -17,7 +18,7 @@ def func(xp):
1718
# CuPy returns zero-dimensional array instead of
1819
# returning a scalar value
1920
assert isinstance(c, xp.ndarray)
20-
assert c.dtype == "p"
21+
assert c.dtype == "l"
2122
assert c.shape == ()
2223
return int(c)
2324

@@ -32,7 +33,7 @@ def func(xp):
3233
# CuPy returns zero-dimensional array instead of
3334
# returning a scalar value
3435
assert isinstance(c, xp.ndarray)
35-
assert c.dtype == "p"
36+
assert c.dtype == "l"
3637
assert c.shape == ()
3738
return int(c)
3839

0 commit comments

Comments
 (0)