Skip to content

Commit aa00062

Browse files
Update dpnp/tests/test_linalg.py
Co-authored-by: Anton <100830759+antonwolfy@users.noreply.github.com>
1 parent 35a6c6e commit aa00062

File tree

1 file changed

+3
-13
lines changed

1 file changed

+3
-13
lines changed

dpnp/tests/test_linalg.py

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -2685,19 +2685,9 @@ def test_lu_p_indices(self, shape, order, dtype):
26852685
A_cast = a_dp.astype(L.dtype, copy=False)
26862686
assert_allclose(A_rec, dpnp.asnumpy(A_cast), rtol=1e-6, atol=1e-6)
26872687

2688-
@pytest.mark.parametrize(
2689-
"in_dtype, expected_p_dtype",
2690-
[
2691-
(dpnp.float32, dpnp.float32),
2692-
(dpnp.float64, dpnp.float64),
2693-
(dpnp.complex64, dpnp.float32),
2694-
(dpnp.complex128, dpnp.float64),
2695-
],
2696-
)
2697-
def test_p_matrix_dtype(self, in_dtype, expected_p_dtype):
2698-
if in_dtype in (dpnp.float64, dpnp.complex128):
2699-
if not has_support_aspect64():
2700-
pytest.skip("fp64 not supported on this device")
2688+
@pytest.mark.parametrize("in_dtype", get_float_complex_dtypes())
2689+
def test_p_matrix_dtype(self, in_dtype):
2690+
expected_p_dtype = numpy.dtype(in_dtype).char.lower()
27012691

27022692
a_np = self._make_nonsingular_np((4, 4), in_dtype, "F")
27032693
a_dp = dpnp.array(a_np, order="F")

0 commit comments

Comments
 (0)