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 | |
| 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')
| -rw-r--r-- | numpy/fft/helper.py | 3 | ||||
| -rw-r--r-- | numpy/fft/tests/test_helper.py | 5 |
2 files changed, 3 insertions, 5 deletions
diff --git a/numpy/fft/helper.py b/numpy/fft/helper.py index 3dacd9ee1..927ee1af1 100644 --- a/numpy/fft/helper.py +++ b/numpy/fft/helper.py @@ -2,7 +2,6 @@ Discrete Fourier Transforms - helper.py """ -from numpy.compat import integer_types from numpy.core import integer, empty, arange, asarray, roll from numpy.core.overrides import array_function_dispatch, set_module @@ -10,7 +9,7 @@ from numpy.core.overrides import array_function_dispatch, set_module __all__ = ['fftshift', 'ifftshift', 'fftfreq', 'rfftfreq'] -integer_types = integer_types + (integer,) +integer_types = (int, integer) def _fftshift_dispatcher(x, axes=None): 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: |