Add check to skip mkl_fft on darwin

Author: rohanbabbar04

pytests/test_ffts.py

@@ -1,5 +1,6 @@
 import itertools
 import os
+import sys
 
 if int(os.environ.get("TEST_CUPY_PYLOPS", 0)):
     import cupy as np
@@ -322,6 +323,8 @@ def test_unknown_engine(par):
 
 @pytest.mark.parametrize("par", pars_fft_small_real)
 def test_FFT_small_real(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
 
     if backend == "numpy" or (backend == "cupy" and par["engine"] == "numpy"):
@@ -399,6 +402,8 @@ def test_FFT_small_real(par):
 )
 @pytest.mark.parametrize("par", pars_fft_random_real)
 def test_FFT_random_real(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
 
     shape = par["shape"]
@@ -455,6 +460,8 @@ def test_FFT_random_real(par):
 
 @pytest.mark.parametrize("par", pars_fft_small_cpx)
 def test_FFT_small_complex(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     dtype, decimal = par["dtype_precision"]
     norm = par["norm"]
@@ -527,6 +534,8 @@ def test_FFT_small_complex(par):
 
 @pytest.mark.parametrize("par", pars_fft_random_cpx)
 def test_FFT_random_complex(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     if backend == "numpy" or (backend == "cupy" and par["engine"] == "numpy"):
         shape = par["shape"]
@@ -614,6 +623,8 @@ def test_FFT_random_complex(par):
 )
 @pytest.mark.parametrize("par", pars_fft2d_random_real)
 def test_FFT2D_random_real(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     if backend == "numpy" or (backend == "cupy" and par["engine"] == "numpy"):
         shape = par["shape"]
@@ -675,6 +686,8 @@ def test_FFT2D_random_real(par):
 
 @pytest.mark.parametrize("par", pars_fft2d_random_cpx)
 def test_FFT2D_random_complex(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     if backend == "numpy" or (backend == "cupy" and par["engine"] == "numpy"):
         shape = par["shape"]
@@ -756,6 +769,8 @@ def test_FFT2D_random_complex(par):
 
 @pytest.mark.parametrize("par", pars_fftnd_random_real)
 def test_FFTND_random_real(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     if backend == "numpy" or (backend == "cupy" and par["engine"] == "numpy"):
         shape = par["shape"]
@@ -817,6 +832,8 @@ def test_FFTND_random_real(par):
 
 @pytest.mark.parametrize("par", pars_fftnd_random_cpx)
 def test_FFTND_random_complex(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     shape = par["shape"]
     dtype, decimal = par["dtype_precision"]
@@ -893,6 +910,8 @@ def test_FFTND_random_complex(par):
 
 @pytest.mark.parametrize("par", pars_fft2dnd_small_cpx)
 def test_FFT2D_small_complex(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     dtype, decimal = par["dtype_precision"]
     norm = par["norm"]
@@ -943,6 +962,8 @@ def test_FFT2D_small_complex(par):
 
 @pytest.mark.parametrize("par", pars_fft2dnd_small_cpx)
 def test_FFTND_small_complex(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     dtype, decimal = par["dtype_precision"]
     norm = par["norm"]
@@ -1018,6 +1039,8 @@ def test_FFTND_small_complex(par):
     ],
 )
 def test_FFT_1dsignal(par):
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     """Dot-test and inversion for FFT operator for 1d signal"""
     decimal = 3 if np.real(np.ones(1, par["dtype"])).dtype == np.float32 else 8
@@ -1135,6 +1158,8 @@ def test_FFT_2dsignal(par):
     """Dot-test and inversion for fft operator for 2d signal
     (fft on single dimension)
     """
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     decimal = 3 if np.real(np.ones(1, par["dtype"])).dtype == np.float32 else 8
 
@@ -1350,6 +1375,8 @@ def test_FFT_3dsignal(par):
     """Dot-test and inversion for fft operator for 3d signal
     (fft on single dimension)
     """
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     decimal = 3 if np.real(np.ones(1, par["dtype"])).dtype == np.float32 else 8
 
@@ -1576,6 +1603,8 @@ def test_FFT_3dsignal(par):
 )
 def test_FFT2D(par):
     """Dot-test and inversion for FFT2D operator for 2d signal"""
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     decimal = 3 if np.real(np.ones(1, par["dtype"])).dtype == np.float32 else 8
 
@@ -1711,6 +1740,8 @@ def test_FFT2D(par):
 )
 def test_FFT3D(par):
     """Dot-test and inversion for FFTND operator for 3d signal"""
+    if par["engine"] == "mkl_fft" and sys.platform == "darwin":
+        pytest.skip("mkl_fft not supported on macOS")
     np.random.seed(5)
     decimal = 3 if np.real(np.ones(1, par["dtype"])).dtype == np.float32 else 8