diff options
| author | Eric Wieser <wieser.eric@gmail.com> | 2020-08-31 10:13:58 +0100 |
|---|---|---|
| committer | Eric Wieser <wieser.eric@gmail.com> | 2020-08-31 10:13:58 +0100 |
| commit | 6e00aaedc40292933c0e8f162d0d4512d6ba350e (patch) | |
| tree | bb3d3aa9926cfd5230d479daf129f1d68b2e231b /numpy/fft/tests/test_helper.py | |
| parent | bbe2cca1925873dea538674b96b53b5cef0a148a (diff) | |
| download | numpy-6e00aaedc40292933c0e8f162d0d4512d6ba350e.tar.gz | |
MAINT: Remove users of `numpy.compat.integer_types`
Some more Python 2 cleanup
Diffstat (limited to 'numpy/fft/tests/test_helper.py')
| -rw-r--r-- | numpy/fft/tests/test_helper.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/numpy/fft/tests/test_helper.py b/numpy/fft/tests/test_helper.py index 68f5990af..3fb700bb3 100644 --- a/numpy/fft/tests/test_helper.py +++ b/numpy/fft/tests/test_helper.py @@ -85,7 +85,6 @@ class TestFFTShift: def test_equal_to_original(self): """ Test that the new (>=v1.15) implementation (see #10073) is equal to the original (<=v1.14) """ - from numpy.compat import integer_types from numpy.core import asarray, concatenate, arange, take def original_fftshift(x, axes=None): @@ -94,7 +93,7 @@ class TestFFTShift: ndim = tmp.ndim if axes is None: axes = list(range(ndim)) - elif isinstance(axes, integer_types): + elif isinstance(axes, int): axes = (axes,) y = tmp for k in axes: @@ -110,7 +109,7 @@ class TestFFTShift: ndim = tmp.ndim if axes is None: axes = list(range(ndim)) - elif isinstance(axes, integer_types): + elif isinstance(axes, int): axes = (axes,) y = tmp for k in axes: |