Skip to content

Commit 452dd74

Browse files
Use assert_dtype_allclose to TestSolve
1 parent a9fa229 commit 452dd74

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dpnp/tests/test_linalg.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2565,7 +2565,7 @@ def test_solve(self, dtype):
25652565
expected = numpy.linalg.solve(a_np, a_np)
25662566
result = dpnp.linalg.solve(a_dp, a_dp)
25672567

2568-
assert_allclose(result, expected)
2568+
assert_dtype_allclose(result, expected)
25692569

25702570
@testing.with_requires("numpy>=2.0")
25712571
@pytest.mark.parametrize("dtype", get_float_complex_dtypes())
@@ -2638,12 +2638,12 @@ def test_solve_strides(self):
26382638
# positive strides
26392639
expected = numpy.linalg.solve(a_np[::2, ::2], b_np[::2])
26402640
result = dpnp.linalg.solve(a_dp[::2, ::2], b_dp[::2])
2641-
assert_allclose(result, expected, rtol=1e-6)
2641+
assert_dtype_allclose(result, expected)
26422642

26432643
# negative strides
26442644
expected = numpy.linalg.solve(a_np[::-2, ::-2], b_np[::-2])
26452645
result = dpnp.linalg.solve(a_dp[::-2, ::-2], b_dp[::-2])
2646-
assert_allclose(result, expected)
2646+
assert_dtype_allclose(result, expected)
26472647

26482648
@pytest.mark.parametrize(
26492649
"matrix, vector",

0 commit comments

Comments
 (0)