diff options
author | Jarrod Millman <millman@berkeley.edu> | 2007-12-29 01:32:27 +0000 |
---|---|---|
committer | Jarrod Millman <millman@berkeley.edu> | 2007-12-29 01:32:27 +0000 |
commit | e7191fc96a676b5ea444f23501a3db61c2be602e (patch) | |
tree | fac59ea48855eb4b0a926303e4f898e5f5643bfa /numpy/fft/tests/test_fftpack.py | |
parent | e31192fd7799a6456459d0e6142ddef8d46b8d08 (diff) | |
download | numpy-e7191fc96a676b5ea444f23501a3db61c2be602e.tar.gz |
janitorial work
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() |