modify test_pyfftw_numpy_interfaces to avoid testing against mkl_fft

s-sajid-ali · Sep 15, 2018 · 119cc88 · 119cc88
1 parent 002bc2f
commit 119cc88
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/test_pyfftw_numpy_interface.py b/test/test_pyfftw_numpy_interface.py
@@ -40,7 +40,12 @@
 from distutils.version import LooseVersion
 import unittest
 import numpy
-from numpy import fft as np_fft
+try:
+    import mkl_fft
+    # mkl_fft monkeypatches numpy.fft so explicitly import from fftpack instead
+    from numpy.fft import fftpack as np_fft
+except ImportError:
+    from numpy import fft as np_fft
 import warnings
 import copy
 warnings.filterwarnings('always')