Skip to content

Commit 012621c

Browse files
modified test files to work with Numpy 1.17.0
1 parent 14dcdf6 commit 012621c

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

mkl_fft/tests/test_fft1d.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,12 @@
3535
import warnings
3636

3737
import mkl_fft
38-
import numpy.fft.fftpack as np_fft
38+
39+
from distutils.version import LooseVersion
40+
if LooseVersion(np.__version__) < LooseVersion('1.17.0'):
41+
import numpy.fft.fftpack as np_fft
42+
else:
43+
import numpy.fft.pocketfft as np_fft
3944

4045

4146
def _datacopied(arr, original):

mkl_fft/tests/test_fftnd.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,12 @@
3535
import warnings
3636

3737
import mkl_fft
38-
import numpy.fft.fftpack as np_fft
38+
39+
from distutils.version import LooseVersion
40+
if LooseVersion(np.__version__) < LooseVersion('1.17.0'):
41+
import numpy.fft.fftpack as np_fft
42+
else:
43+
import numpy.fft.pocketfft as np_fft
3944

4045
reps_64 = (2**11)*np.finfo(np.float64).eps
4146
reps_32 = (2**11)*np.finfo(np.float32).eps

0 commit comments

Comments
 (0)