Skip to content

Commit 28aedbc

Browse files
committed
Assert correct result dtype in test_count.py
1 parent 2c9855e commit 28aedbc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/third_party/cupy/sorting_tests/test_count.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def func(xp):
1717
# CuPy returns zero-dimensional array instead of
1818
# returning a scalar value
1919
assert isinstance(c, xp.ndarray)
20-
assert c.dtype == "l"
20+
assert c.dtype == "p"
2121
assert c.shape == ()
2222
return int(c)
2323

@@ -32,7 +32,7 @@ def func(xp):
3232
# CuPy returns zero-dimensional array instead of
3333
# returning a scalar value
3434
assert isinstance(c, xp.ndarray)
35-
assert c.dtype == "l"
35+
assert c.dtype == "p"
3636
assert c.shape == ()
3737
return int(c)
3838

0 commit comments

Comments
 (0)