diff options
Diffstat (limited to 'numpy/fft/tests/test_fftpack.py')
-rw-r--r-- | numpy/fft/tests/test_fftpack.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/numpy/fft/tests/test_fftpack.py b/numpy/fft/tests/test_fftpack.py index 53d4a8c66..d1409bd69 100644 --- a/numpy/fft/tests/test_fftpack.py +++ b/numpy/fft/tests/test_fftpack.py @@ -1,24 +1,24 @@ import sys from numpy.testing import * set_package_path() -import numpy as N +import numpy as np restore_path() def fft1(x): L = len(x) - phase = -2j*N.pi*(N.arange(L)/float(L)) - phase = N.arange(L).reshape(-1,1) * phase - return N.sum(x*N.exp(phase),axis=1) + phase = -2j*np.pi*(np.arange(L)/float(L)) + phase = np.arange(L).reshape(-1,1) * phase + return np.sum(x*np.exp(phase),axis=1) class TestFFTShift(NumpyTestCase): def check_fft_n(self): - self.failUnlessRaises(ValueError,N.fft.fft,[1,2,3],0) + self.failUnlessRaises(ValueError,np.fft.fft,[1,2,3],0) class TestFFT1D(NumpyTestCase): def check_basic(self): - rand = N.random.random + rand = np.random.random x = rand(30) + 1j*rand(30) - assert_array_almost_equal(fft1(x), N.fft.fft(x)) + assert_array_almost_equal(fft1(x), np.fft.fft(x)) if __name__ == "__main__": NumpyTest().run() |