Update fft tests

Author: antonwolfy

dpnp/tests/third_party/cupy/fft_tests/test_fft.py

@@ -889,7 +889,7 @@ def test_ifftn_orders(self, dtype, enable_nd):
                 pass
 
 
-# @testing.with_requires("numpy>=2.0")
+@testing.with_requires("numpy>=2.0")
 @pytest.mark.usefixtures("skip_forward_backward")
 @testing.parameterize(
     *testing.product(
@@ -925,17 +925,6 @@ def test_rfft(self, xp, dtype):
     def test_irfft(self, xp, dtype):
         a = testing.shaped_random(self.shape, xp, dtype)
         out = xp.fft.irfft(a, n=self.n, norm=self.norm)
-
-        if dtype == xp.float16 and xp is cupy:
-            # XXX: np2.0: f16 dtypes differ
-            out = out.astype(np.float16)
-        elif (
-            xp is np
-            and np.lib.NumpyVersion(np.__version__) < "2.0.0"
-            and dtype == np.float32
-        ):
-            out = out.astype(np.float32)
-
         return out
 
 
@@ -1008,7 +997,7 @@ def test_rfft_error_on_wrong_plan(self, dtype):
         assert "Target array size does not match the plan." in str(ex.value)
 
 
-# @testing.with_requires("numpy>=2.0")
+@testing.with_requires("numpy>=2.0")
 @pytest.mark.usefixtures("skip_forward_backward")
 @testing.parameterize(
     *(
@@ -1069,13 +1058,6 @@ def test_irfft2(self, xp, dtype, order, enable_nd):
 
         if self.s is None and self.axes in [None, (-2, -1)]:
             pytest.skip("Input is not Hermitian Symmetric")
-        elif dtype == xp.float16 and xp is cupy:
-            pytest.xfail("XXX: np2.0: f16 dtypes differ")
-        elif (
-            np.lib.NumpyVersion(np.__version__) < "2.0.0"
-            and dtype == np.float32
-        ):
-            pytest.skip("dtypes differ")
 
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
@@ -1105,7 +1087,7 @@ def test_irfft2(self, dtype):
                 xp.fft.irfft2(a, s=self.s, axes=self.axes, norm=self.norm)
 
 
-# @testing.with_requires("numpy>=2.0")
+@testing.with_requires("numpy>=2.0")
 @pytest.mark.usefixtures("skip_forward_backward")
 @testing.parameterize(
     *(
@@ -1166,13 +1148,6 @@ def test_irfftn(self, xp, dtype, order, enable_nd):
 
         if self.s is None and self.axes in [None, (-2, -1)]:
             pytest.skip("Input is not Hermitian Symmetric")
-        elif dtype == xp.float16 and xp is cupy:
-            pytest.xfail("XXX: np2.0: f16 dtypes differ")
-        elif (
-            np.lib.NumpyVersion(np.__version__) < "2.0.0"
-            and dtype == np.float32
-        ):
-            pytest.skip("dtypes differ")
 
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
@@ -1243,10 +1218,6 @@ def test_rfftn(self, xp, dtype, enable_nd):
     def test_irfftn(self, xp, dtype, enable_nd):
         assert config.enable_nd_planning == enable_nd
         a = testing.shaped_random(self.shape, xp, dtype)
-
-        if dtype == xp.float16 and xp is cupy:
-            pytest.xfail("XXX: np2.0: f16 dtypes differ")
-
         if xp is np:
             return xp.fft.irfftn(a, s=self.s, axes=self.axes, norm=self.norm)
 
@@ -1349,7 +1320,7 @@ def test_irfftn(self, dtype):
                 xp.fft.irfftn(a, s=self.s, axes=self.axes, norm=self.norm)
 
 
-# @testing.with_requires("numpy>=2.0")
+@testing.with_requires("numpy>=2.0")
 @pytest.mark.usefixtures("skip_forward_backward")
 @testing.parameterize(
     *testing.product(
@@ -1373,17 +1344,6 @@ class TestHfft:
     def test_hfft(self, xp, dtype):
         a = testing.shaped_random(self.shape, xp, dtype)
         out = xp.fft.hfft(a, n=self.n, norm=self.norm)
-
-        if dtype == xp.float16 and xp is cupy:
-            # XXX: np2.0: f16 dtypes differ
-            out = out.astype(np.float16)
-        elif (
-            xp is np
-            and np.lib.NumpyVersion(np.__version__) < "2.0.0"
-            and dtype == np.float32
-        ):
-            out = out.astype(np.float32)
-
         return out
 
     @testing.for_all_dtypes(no_complex=True)
@@ -1396,16 +1356,7 @@ def test_hfft(self, xp, dtype):
     )
     def test_ihfft(self, xp, dtype):
         a = testing.shaped_random(self.shape, xp, dtype)
-        out = xp.fft.ihfft(a, n=self.n, norm=self.norm)
-
-        if (
-            xp is np
-            and np.lib.NumpyVersion(np.__version__) < "2.0.0"
-            and dtype == np.float32
-        ):
-            out = out.astype(np.complex64)
-
-        return out
+        return xp.fft.ihfft(a, n=self.n, norm=self.norm)
 
 
 # @testing.with_requires("numpy>=2.0")