diff options
Diffstat (limited to 'numpy/fft')
| -rw-r--r-- | numpy/fft/fftpack.py | 2 | ||||
| -rw-r--r-- | numpy/fft/setup.py | 1 | ||||
| -rw-r--r-- | numpy/fft/tests/test_helper.py | 4 |
3 files changed, 3 insertions, 4 deletions
diff --git a/numpy/fft/fftpack.py b/numpy/fft/fftpack.py index 706fcdd2f..4efb2a9a0 100644 --- a/numpy/fft/fftpack.py +++ b/numpy/fft/fftpack.py @@ -43,7 +43,7 @@ _fft_cache = {} _real_fft_cache = {} def _raw_fft(a, n=None, axis=-1, init_function=fftpack.cffti, - work_function=fftpack.cfftf, fft_cache = _fft_cache ): + work_function=fftpack.cfftf, fft_cache=_fft_cache): a = asarray(a) if n is None: diff --git a/numpy/fft/setup.py b/numpy/fft/setup.py index 79f681e55..cd99a82d7 100644 --- a/numpy/fft/setup.py +++ b/numpy/fft/setup.py @@ -12,7 +12,6 @@ def configuration(parent_package='',top_path=None): sources=['fftpack_litemodule.c', 'fftpack.c'] ) - return config if __name__ == '__main__': diff --git a/numpy/fft/tests/test_helper.py b/numpy/fft/tests/test_helper.py index 7eaa99fdb..1a51f8e3a 100644 --- a/numpy/fft/tests/test_helper.py +++ b/numpy/fft/tests/test_helper.py @@ -30,8 +30,8 @@ class TestFFTShift(TestCase): assert_array_almost_equal(fft.ifftshift(fft.fftshift(x)), x) def test_axes_keyword(self): - freqs = [[ 0, 1, 2], [ 3, 4, -4], [-3, -2, -1]] - shifted = [[-1, -3, -2], [ 2, 0, 1], [-4, 3, 4]] + freqs = [[0, 1, 2], [3, 4, -4], [-3, -2, -1]] + shifted = [[-1, -3, -2], [2, 0, 1], [-4, 3, 4]] assert_array_almost_equal(fft.fftshift(freqs, axes=(0, 1)), shifted) assert_array_almost_equal(fft.fftshift(freqs, axes=0), fft.fftshift(freqs, axes=(0,))) |