Update test_sycl_queue

Author: vlad-perevezentsev

tests/test_sycl_queue.py

@@ -1663,20 +1663,26 @@ def test_slogdet(shape, is_empty, device):
 
 
 @pytest.mark.parametrize(
-    "shape, hermitian",
+    "shape, hermitian, rcond_as_array",
     [
-        ((4, 4), False),
-        ((2, 0), False),
-        ((4, 4), True),
-        ((2, 2, 3), False),
-        ((0, 2, 3), False),
-        ((1, 0, 3), False),
+        ((4, 4), False, False),
+        ((4, 4), False, True),
+        ((2, 0), False, False),
+        ((4, 4), True, False),
+        ((4, 4), True, True),
+        ((2, 2, 3), False, False),
+        ((2, 2, 3), False, True),
+        ((0, 2, 3), False, False),
+        ((1, 0, 3), False, False),
     ],
     ids=[
         "(4, 4)",
+        "(4, 4), rcond_as_array",
         "(2, 0)",
         "(2, 2), hermitian)",
+        "(2, 2), hermitian, rcond_as_array)",
         "(2, 2, 3)",
+        "(2, 2, 3), rcond_as_array",
         "(0, 2, 3)",
         "(1, 0, 3)",
     ],
@@ -1686,7 +1692,7 @@ def test_slogdet(shape, is_empty, device):
     valid_devices,
     ids=[device.filter_string for device in valid_devices],
 )
-def test_pinv(shape, hermitian, device):
+def test_pinv(shape, hermitian, rcond_as_array, device):
     if hermitian:
         a_np = numpy.random.randn(*shape) + 1j * numpy.random.randn(*shape)
         a_np = numpy.conj(a_np.T) @ a_np
@@ -1695,8 +1701,20 @@ def test_pinv(shape, hermitian, device):
 
     a_dp = dpnp.array(a_np, device=device)
 
-    B_result = dpnp.linalg.pinv(a_dp, hermitian=hermitian)
-    B_expected = numpy.linalg.pinv(a_np, hermitian=hermitian)
+    if rcond_as_array:
+        rcond_np = numpy.array(1e-15)
+        rcond_dp = dpnp.array(1e-15, device=device)
+
+        B_result = dpnp.linalg.pinv(a_dp, rcond=rcond_dp, hermitian=hermitian)
+        B_expected = numpy.linalg.pinv(
+            a_np, rcond=rcond_np, hermitian=hermitian
+        )
+
+    else:
+        # rcond == 1e-15 by default
+        B_result = dpnp.linalg.pinv(a_dp, hermitian=hermitian)
+        B_expected = numpy.linalg.pinv(a_np, hermitian=hermitian)
+
     assert_allclose(B_expected, B_result, rtol=1e-3, atol=1e-4)
 
     B_queue = B_result.sycl_queue