diff options
Diffstat (limited to 'numpy/fft/tests/test_fftpack.py')
-rw-r--r-- | numpy/fft/tests/test_fftpack.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/numpy/fft/tests/test_fftpack.py b/numpy/fft/tests/test_fftpack.py index 74e2d06f5..53d4a8c66 100644 --- a/numpy/fft/tests/test_fftpack.py +++ b/numpy/fft/tests/test_fftpack.py @@ -1,12 +1,24 @@ import sys from numpy.testing import * set_package_path() -from numpy.fft import * +import numpy as N 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) + class TestFFTShift(NumpyTestCase): def check_fft_n(self): - self.failUnlessRaises(ValueError,fft,[1,2,3],0) + self.failUnlessRaises(ValueError,N.fft.fft,[1,2,3],0) + +class TestFFT1D(NumpyTestCase): + def check_basic(self): + rand = N.random.random + x = rand(30) + 1j*rand(30) + assert_array_almost_equal(fft1(x), N.fft.fft(x)) if __name__ == "__main__": NumpyTest().run() |