tests: don't require fftw for tests

steven-murray · Apr 30, 2024 · d2bc14d · d2bc14d
1 parent 69d6c44
commit d2bc14d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_fft.py b/tests/test_fft.py
@@ -14,15 +14,19 @@
     (1, 1, 0, 1),
 ]
 
-BACKENDS = [NumpyFFT(), FFTW(nthreads=1)]
+BACKENDS = [
+    NumpyFFT(),
+]
 
 try:
     import pyfftw
 
+    BACKENDS.append(FFTW(nthreads=1))
+
     pyfftw.builders._utils._default_threads(4)
 
     BACKENDS.append(FFTW(nthreads=2))
-except ValueError:
+except (ValueError, ImportError):
     # If FFTW was not installed with multithreading, the above will error.
     pass