Add CFD tests

Author: antonwolfy

dpnp/tests/test_sycl_queue.py

@@ -505,6 +505,34 @@ def test_2in_1out(func, data1, data2, device):
         pass
 
 
+@pytest.mark.parametrize(
+    "func, data1, data2",
+    [
+        pytest.param("divmod", numpy.arange(5), numpy.array(3)),
+    ],
+)
+@pytest.mark.parametrize("device", valid_dev, ids=dev_ids)
+def test_2in_2out(func, data1, data2, device):
+    x1 = dpnp.array(data1, device=device)
+    x2 = dpnp.array(data2, device=device)
+    res1, res2 = getattr(dpnp, func)(x1, x2)
+
+    assert_sycl_queue_equal(res1.sycl_queue, res2.sycl_queue)
+    assert_sycl_queue_equal(res1.sycl_queue, x1.sycl_queue)
+    assert_sycl_queue_equal(res1.sycl_queue, x2.sycl_queue)
+
+    out1 = dpnp.empty_like(res1)
+    out2 = dpnp.empty_like(res2)
+    try:
+        # some functions do not support out kwarg
+        getattr(dpnp, func)(x1, x2, out=(out1, out2))
+        assert_sycl_queue_equal(out1.sycl_queue, out2.sycl_queue)
+        assert_sycl_queue_equal(out1.sycl_queue, x1.sycl_queue)
+        assert_sycl_queue_equal(out1.sycl_queue, x2.sycl_queue)
+    except TypeError:
+        pass
+
+
 @pytest.mark.parametrize(
     "op",
     [

dpnp/tests/test_usm_type.py

@@ -667,7 +667,7 @@ def test_1in_2out(func, data, usm_type):
 
 
 @pytest.mark.parametrize(
-    "func,data1,data2",
+    "func, data1, data2",
     [
         pytest.param(
             "allclose",
@@ -742,6 +742,28 @@ def test_2in_1out(func, data1, data2, usm_type_x, usm_type_y):
     assert z.usm_type == du.get_coerced_usm_type([usm_type_x, usm_type_y])
 
 
+@pytest.mark.parametrize(
+    "func, data1, data2",
+    [
+        pytest.param("divmod", numpy.arange(5), numpy.array(3)),
+    ],
+)
+@pytest.mark.parametrize("usm_type_x", list_of_usm_types)
+@pytest.mark.parametrize("usm_type_y", list_of_usm_types)
+def test_2in_2out(func, data1, data2, usm_type_x, usm_type_y):
+    x = dpnp.array(data1, usm_type=usm_type_x)
+    y = dpnp.array(data2, usm_type=usm_type_y)
+    z1, z2 = getattr(dpnp, func)(x, y)
+
+    assert x.usm_type == usm_type_x
+    assert y.usm_type == usm_type_y
+    assert (
+        z1.usm_type
+        == z2.usm_type
+        == du.get_coerced_usm_type([usm_type_x, usm_type_y])
+    )
+
+
 @pytest.mark.parametrize(
     "func, data, scalar",
     [