add new tests

Author: Vahid Tavanashad

tests/test_sycl_queue.py

@@ -2048,18 +2048,34 @@ def test_concat_stack(func, data1, data2, device):
     assert_sycl_queue_equal(result.sycl_queue, x2.sycl_queue)
 
 
-@pytest.mark.parametrize(
-    "device",
-    valid_devices,
-    ids=[device.filter_string for device in valid_devices],
-)
-@pytest.mark.parametrize(
-    "obj", [slice(None, None, 2), 3, [2, 3]], ids=["slice", "int", "list"]
-)
-def test_delete(device, obj):
-    x = dpnp.arange(5, device=device)
-    result = dpnp.delete(x, obj)
-    assert_sycl_queue_equal(result.sycl_queue, x.sycl_queue)
+class TestDelete:
+    @pytest.mark.parametrize(
+        "obj",
+        [slice(None, None, 2), 3, [2, 3]],
+        ids=["slice", "scalar", "list"],
+    )
+    @pytest.mark.parametrize(
+        "device",
+        valid_devices,
+        ids=[device.filter_string for device in valid_devices],
+    )
+    def test_delete(self, obj, device):
+        x = dpnp.arange(5, device=device)
+        result = dpnp.delete(x, obj)
+        assert_sycl_queue_equal(result.sycl_queue, x.sycl_queue)
+
+    @pytest.mark.parametrize(
+        "device",
+        valid_devices,
+        ids=[device.filter_string for device in valid_devices],
+    )
+    def test_obj_ndarray(self, device):
+        x = dpnp.arange(5, device=device)
+        y = dpnp.array([1, 4], device=device)
+        result = dpnp.delete(x, y)
+
+        assert_sycl_queue_equal(result.sycl_queue, x.sycl_queue)
+        assert_sycl_queue_equal(result.sycl_queue, y.sycl_queue)
 
 
 @pytest.mark.parametrize(

tests/test_usm_type.py

@@ -831,16 +831,36 @@ def test_cond(usm_type, p):
     assert result.usm_type == usm_type
 
 
-@pytest.mark.parametrize("usm_type", list_of_usm_types, ids=list_of_usm_types)
-@pytest.mark.parametrize(
-    "obj", [slice(None, None, 2), 3, [2, 3]], ids=["slice", "int", "list"]
-)
-def test_delete(usm_type, obj):
-    x = dp.arange(5, usm_type=usm_type)
-    result = dp.delete(x, obj)
+class TestDelete:
+    @pytest.mark.parametrize(
+        "obj",
+        [slice(None, None, 2), 3, [2, 3]],
+        ids=["slice", "scalar", "list"],
+    )
+    @pytest.mark.parametrize(
+        "usm_type", list_of_usm_types, ids=list_of_usm_types
+    )
+    def test_delete(self, obj, usm_type):
+        x = dp.arange(5, usm_type=usm_type)
+        result = dp.delete(x, obj)
 
-    assert x.usm_type == usm_type
-    assert result.usm_type == usm_type
+        assert x.usm_type == usm_type
+        assert result.usm_type == usm_type
+
+    @pytest.mark.parametrize(
+        "usm_type_x", list_of_usm_types, ids=list_of_usm_types
+    )
+    @pytest.mark.parametrize(
+        "usm_type_y", list_of_usm_types, ids=list_of_usm_types
+    )
+    def test_obj_ndarray(self, usm_type_x, usm_type_y):
+        x = dp.arange(5, usm_type=usm_type_x)
+        y = dp.array([1, 4], usm_type=usm_type_y)
+        z = dp.delete(x, y)
+
+        assert x.usm_type == usm_type_x
+        assert y.usm_type == usm_type_y
+        assert z.usm_type == du.get_coerced_usm_type([usm_type_x, usm_type_y])
 
 
 @pytest.mark.parametrize("usm_type", list_of_usm_types, ids=list_of_usm_types)