Add test_qr in test_usm_type

Author: vlad-perevezentsev

tests/test_sycl_queue.py

@@ -1201,24 +1201,22 @@ def test_matrix_rank(device):
 def test_qr(shape, mode, device):
     dtype = dpnp.default_float_type(device)
     count_elems = numpy.prod(shape)
-    dpnp_data = dpnp.arange(count_elems, dtype=dtype, device=device).reshape(
-        shape
-    )
+    a = dpnp.arange(count_elems, dtype=dtype, device=device).reshape(shape)
 
-    expected_queue = dpnp_data.get_array().sycl_queue
+    expected_queue = a.get_array().sycl_queue
 
     if mode == "r":
-        dpnp_r = dpnp.linalg.qr(dpnp_data, mode=mode)
-        dpnp_r_queue = dpnp_r.get_array().sycl_queue
-        assert_sycl_queue_equal(dpnp_r_queue, expected_queue)
+        dp_r = dpnp.linalg.qr(a, mode=mode)
+        dp_r_queue = dp_r.get_array().sycl_queue
+        assert_sycl_queue_equal(dp_r_queue, expected_queue)
     else:
-        dpnp_q, dpnp_r = dpnp.linalg.qr(dpnp_data, mode=mode)
+        dp_q, dp_r = dpnp.linalg.qr(a, mode=mode)
 
-        dpnp_q_queue = dpnp_q.get_array().sycl_queue
-        dpnp_r_queue = dpnp_r.get_array().sycl_queue
+        dp_q_queue = dp_q.get_array().sycl_queue
+        dp_r_queue = dp_r.get_array().sycl_queue
 
-        assert_sycl_queue_equal(dpnp_q_queue, expected_queue)
-        assert_sycl_queue_equal(dpnp_r_queue, expected_queue)
+        assert_sycl_queue_equal(dp_q_queue, expected_queue)
+        assert_sycl_queue_equal(dp_r_queue, expected_queue)
 
 
 @pytest.mark.parametrize(

tests/test_usm_type.py

@@ -710,3 +710,40 @@ def test_det(shape, is_empty, usm_type):
     det = dp.linalg.det(x)
 
     assert x.usm_type == det.usm_type
+
+
+@pytest.mark.parametrize("usm_type", list_of_usm_types, ids=list_of_usm_types)
+@pytest.mark.parametrize(
+    "shape",
+    [
+        (4, 4),
+        (2, 0),
+        (2, 2, 3),
+        (0, 2, 3),
+        (1, 0, 3),
+    ],
+    ids=[
+        "(4, 4)",
+        "(2, 0)",
+        "(2, 2, 3)",
+        "(0, 2, 3)",
+        "(1, 0, 3)",
+    ],
+)
+@pytest.mark.parametrize(
+    "mode",
+    ["r", "raw", "complete", "reduced"],
+    ids=["r", "raw", "complete", "reduced"],
+)
+def test_qr(shape, mode, usm_type):
+    count_elems = numpy.prod(shape)
+    a = dp.arange(count_elems, usm_type=usm_type).reshape(shape)
+
+    if mode == "r":
+        dp_r = dp.linalg.qr(a, mode=mode)
+        assert a.usm_type == dp_r.usm_type
+    else:
+        dp_q, dp_r = dp.linalg.qr(a, mode=mode)
+
+        assert a.usm_type == dp_q.usm_type
+        assert a.usm_type == dp_r.usm_type