Skip to content

Commit 2811654

Browse files
author
Vahid Tavanashad
committed
fix a test
1 parent d25e78c commit 2811654

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

dpnp/tests/test_nanfunctions.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -462,19 +462,15 @@ def test_all_nan(self):
462462
assert_dtype_allclose(result, expected)
463463

464464
@pytest.mark.parametrize("axis", [None, 0, -1, (0, -2, -1)])
465-
def test_overwrite_input(self, axis, keepdims):
465+
def test_overwrite_input(self, axis):
466466
a = numpy.random.uniform(-5, 5, 24).reshape(2, 3, 4)
467467
a[0, 0, 0] = a[-2, -2, -2] = numpy.nan
468468
ia = dpnp.array(a)
469469

470470
b = a.copy()
471471
ib = ia.copy()
472-
expected = numpy.nanmedian(
473-
b, axis=axis, keepdims=keepdims, overwrite_input=True
474-
)
475-
result = dpnp.nanmedian(
476-
ib, axis=axis, keepdims=keepdims, overwrite_input=True
477-
)
472+
expected = numpy.nanmedian(b, axis=axis, overwrite_input=True)
473+
result = dpnp.nanmedian(ib, axis=axis, overwrite_input=True)
478474
assert not numpy.all(a == b)
479475
assert not dpnp.all(ia == ib)
480476

0 commit comments

Comments
 (0)