Skip to content

Commit 706b3e2

Browse files
Run TestQr only for float and complex dtypes
1 parent 26d213a commit 706b3e2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

dpnp/tests/test_linalg.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3205,16 +3205,16 @@ def check_qr(self, a_np, a_dp, mode):
32053205
exp_R = exp_res.R
32063206
assert_allclose(R_dp, exp_R, atol=1e-4, rtol=1e-4)
32073207

3208-
exp_dp = self.gram(a_dp, dpnp).astype(R_dp.dtype)
3209-
exp_np = self.gram(a_np, numpy).astype(R_np.dtype)
3208+
exp_dp = self.gram(a_dp, dpnp)
3209+
exp_np = self.gram(a_np, numpy)
32103210

32113211
# compare R^H @ R == A^H @ A
32123212
assert_allclose(self.gram(R_dp, dpnp), exp_dp, atol=1e-4, rtol=1e-4)
32133213
assert_allclose(
32143214
self.gram(R_np, numpy), exp_np, atol=1e-4, rtol=1e-4
32153215
)
32163216

3217-
@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
3217+
@pytest.mark.parametrize("dtype", get_float_complex_dtypes())
32183218
@pytest.mark.parametrize(
32193219
"shape",
32203220
[
@@ -3247,7 +3247,7 @@ def test_qr(self, dtype, shape, mode):
32473247

32483248
self.check_qr(a, ia, mode)
32493249

3250-
@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
3250+
@pytest.mark.parametrize("dtype", get_float_complex_dtypes())
32513251
@pytest.mark.parametrize(
32523252
"shape",
32533253
[(32, 32), (8, 16, 16)],
@@ -3260,7 +3260,7 @@ def test_qr_large(self, dtype, shape, mode):
32603260

32613261
self.check_qr(a, ia, mode)
32623262

3263-
@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
3263+
@pytest.mark.parametrize("dtype", get_float_complex_dtypes())
32643264
@pytest.mark.parametrize(
32653265
"shape",
32663266
[(0, 0), (0, 2), (2, 0), (2, 0, 3), (2, 3, 0), (0, 2, 3)],

0 commit comments

Comments
 (0)