Skip to content

Commit 607be70

Browse files
committed
Add a test to cover that
1 parent ff4c32a commit 607be70

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

dpnp/tests/test_manipulation.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1838,6 +1838,15 @@ def test_equal_nan(self, eq_nan_kwd):
18381838
expected = numpy.unique(a, **eq_nan_kwd)
18391839
assert_array_equal(result, expected)
18401840

1841+
@testing.with_requires("numpy>=2.3.2")
1842+
def test_1d_equal_nan_axis0(self):
1843+
a = numpy.array([numpy.nan, 0, 0, numpy.nan])
1844+
ia = dpnp.array(a)
1845+
1846+
result = dpnp.unique(ia, axis=0, equal_nan=True)
1847+
expected = numpy.unique(a, axis=0, equal_nan=True)
1848+
assert_array_equal(result, expected)
1849+
18411850
@pytest.mark.parametrize("dt", get_float_complex_dtypes())
18421851
@pytest.mark.parametrize(
18431852
"axis_kwd",

0 commit comments

Comments
 (0)