Skip to content

Commit 1eb1592

Browse files
authored
update test_strides.py file (#2317)
In this PR, `test_strides.py` file has been updated to reduce duplication. In addition, the test for `dpnp.sign` has been updated to better validate cases involving complex numbers.
1 parent 89cb658 commit 1eb1592

File tree

2 files changed

+204
-277
lines changed

2 files changed

+204
-277
lines changed

dpnp/tests/test_mathematical.py

Lines changed: 16 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -2309,35 +2309,27 @@ def test_float_remainder_fmod_nans_inf(func, dtype, lhs, rhs):
23092309
assert_equal(result, expected)
23102310

23112311

2312+
@testing.with_requires("numpy>=2.0.0")
23122313
@pytest.mark.parametrize(
2313-
"data",
2314-
[[2, 0, -2], [1.1, -1.1]],
2315-
ids=["[2, 0, -2]", "[1.1, -1.1]"],
2316-
)
2317-
@pytest.mark.parametrize(
2318-
"dtype", get_all_dtypes(no_bool=True, no_unsigned=True)
2314+
"dtype", get_all_dtypes(no_none=True, no_unsigned=True)
23192315
)
2320-
def test_sign(data, dtype):
2321-
np_a = numpy.array(data, dtype=dtype)
2322-
dpnp_a = dpnp.array(data, dtype=dtype)
2316+
def test_sign(dtype):
2317+
a = generate_random_numpy_array((2, 3), dtype=dtype)
2318+
ia = dpnp.array(a, dtype=dtype)
23232319

2324-
result = dpnp.sign(dpnp_a)
2325-
expected = numpy.sign(np_a)
2326-
assert_dtype_allclose(result, expected)
2327-
2328-
# out keyword
2329-
if dtype is not None:
2330-
dp_out = dpnp.empty(expected.shape, dtype=expected.dtype)
2331-
result = dpnp.sign(dpnp_a, out=dp_out)
2332-
assert dp_out is result
2320+
if dtype == dpnp.bool:
2321+
assert_raises(TypeError, dpnp.sign, ia)
2322+
assert_raises(TypeError, numpy.sign, a)
2323+
else:
2324+
result = dpnp.sign(ia)
2325+
expected = numpy.sign(a)
23332326
assert_dtype_allclose(result, expected)
23342327

2335-
2336-
def test_sign_boolean():
2337-
dpnp_a = dpnp.array([True, False])
2338-
2339-
with pytest.raises(TypeError):
2340-
dpnp.sign(dpnp_a)
2328+
# out keyword
2329+
iout = dpnp.empty(expected.shape, dtype=expected.dtype)
2330+
result = dpnp.sign(ia, out=iout)
2331+
assert iout is result
2332+
assert_dtype_allclose(result, expected)
23412333

23422334

23432335
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)