update tests

Author: vtavana

tests/test_sycl_queue.py

@@ -1135,21 +1135,21 @@ def test_out_multi_dot(device):
         assert_sycl_queue_equal(result.sycl_queue, exec_q)
 
 
-@pytest.mark.parametrize("type", ["complex128"])
+@pytest.mark.parametrize("func", ["fft", "ifft"])
 @pytest.mark.parametrize(
     "device",
     valid_devices,
     ids=[device.filter_string for device in valid_devices],
 )
-def test_fft(type, device):
-    data = numpy.arange(100, dtype=numpy.dtype(type))
+def test_fft(func, device):
+    data = numpy.arange(100, dtype=numpy.complex128)
 
     dpnp_data = dpnp.array(data, device=device)
 
-    expected = numpy.fft.fft(data)
-    result = dpnp.fft.fft(dpnp_data)
+    expected = getattr(numpy.fft, func)(data)
+    result = getattr(dpnp.fft, func)(dpnp_data)
 
-    assert_allclose(result, expected, rtol=1e-4, atol=1e-7)
+    assert_dtype_allclose(result, expected)
 
     expected_queue = dpnp_data.get_array().sycl_queue
     result_queue = result.get_array().sycl_queue

tests/test_usm_type.py

@@ -914,6 +914,17 @@ def test_eigenvalue(func, shape, usm_type):
     assert a.usm_type == dp_val.usm_type
 
 
+@pytest.mark.parametrize("func", ["fft", "ifft"])
+@pytest.mark.parametrize("usm_type", list_of_usm_types, ids=list_of_usm_types)
+def test_fft(func, usm_type):
+    data = numpy.arange(100, dtype=numpy.complex128)
+    dpnp_data = dp.array(data, usm_type=usm_type)
+    result = getattr(dp.fft, func)(dpnp_data)
+
+    assert dpnp_data.usm_type == usm_type
+    assert result.usm_type == usm_type
+
+
 @pytest.mark.parametrize(
     "usm_type_matrix", list_of_usm_types, ids=list_of_usm_types
 )