Increase test code coverage. Add documentation.

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()