Add USM type and SYCL queue tests for compress

Author: ndgrigorian

dpnp/tests/test_sycl_queue.py

@@ -2855,3 +2855,24 @@ def test_ix(device_0, device_1):
     ixgrid = dpnp.ix_(x0, x1)
     assert_sycl_queue_equal(ixgrid[0].sycl_queue, x0.sycl_queue)
     assert_sycl_queue_equal(ixgrid[1].sycl_queue, x1.sycl_queue)
+
+
+@pytest.mark.parametrize(
+    "device",
+    valid_devices,
+    ids=[device.filter_string for device in valid_devices],
+)
+def test_compress(device):
+    a_np = numpy.arange(5)
+    a = dpnp.array(a_np, device=device)
+
+    cond_np = numpy.array([0, 1, 0])
+    cond = dpnp.array(cond_np, device=device)
+
+    expected = numpy.compress(cond_np, a_np, axis=None)
+    result = dpnp.compress(cond, a, axis=None)
+    assert_allclose(expected, result)
+
+    expected_queue = a.sycl_queue
+    result_queue = result.sycl_queue
+    assert_sycl_queue_equal(result_queue, expected_queue)

dpnp/tests/test_usm_type.py

@@ -1746,3 +1746,17 @@ def test_ix(usm_type_0, usm_type_1):
     ixgrid = dp.ix_(x0, x1)
     assert ixgrid[0].usm_type == x0.usm_type
     assert ixgrid[1].usm_type == x1.usm_type
+
+
+@pytest.mark.parametrize("usm_type_a", list_of_usm_types, ids=list_of_usm_types)
+@pytest.mark.parametrize(
+    "usm_type_cond", list_of_usm_types, ids=list_of_usm_types
+)
+def test_compress(usm_type_a, usm_type_cond):
+    a = dp.arange(5, usm_type=usm_type_a)
+    cond = dp.array([False, True, True], usm_type=usm_type_cond)
+    z = dp.compress(cond, a, axis=None)
+
+    assert a.usm_type == usm_type_a
+    assert cond.usm_type == usm_type_cond
+    assert z.usm_type == du.get_coerced_usm_type([usm_type_a, usm_type_cond])