2 files changed, 57 insertions, 0 deletions
diff --git a/numpy/fft/tests/test_fftpack.py b/numpy/fft/tests/test_fftpack.py
new file mode 100644
index 000000000..0e38fb3ea
--- /dev/null
+++ b/numpy/fft/tests/test_fftpack.py
@@ -0,0 +1,12 @@
+import sys
+from numpy.testing import *
+set_package_path()
+from numpy.fft import *
+restore_path()
+
+class test_fftshift(NumpyTestCase):
+    def check_fft_n(self):
+        self.failUnlessRaises(ValueError,fft,[1,2,3],0)
+
+if __name__ == "__main__":
+    NumpyTest().run()
diff --git a/numpy/fft/tests/test_helper.py b/numpy/fft/tests/test_helper.py
new file mode 100644
index 000000000..3d02c01df
--- /dev/null
+++ b/numpy/fft/tests/test_helper.py
@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+# Copied from fftpack.helper by Pearu Peterson, October 2005
+""" Test functions for fftpack.helper module
+"""
+
+import sys
+from numpy.testing import *
+set_package_path()
+from numpy.fft import fftshift,ifftshift,fftfreq
+del sys.path[0]
+
+from numpy import pi
+
+def random(size):
+    return rand(*size)
+
+class test_fftshift(NumpyTestCase):
+
+    def check_definition(self):
+        x = [0,1,2,3,4,-4,-3,-2,-1]
+        y = [-4,-3,-2,-1,0,1,2,3,4]
+        assert_array_almost_equal(fftshift(x),y)
+        assert_array_almost_equal(ifftshift(y),x)
+        x = [0,1,2,3,4,-5,-4,-3,-2,-1]
+        y = [-5,-4,-3,-2,-1,0,1,2,3,4]
+        assert_array_almost_equal(fftshift(x),y)
+        assert_array_almost_equal(ifftshift(y),x)
+
+    def check_inverse(self):
+        for n in [1,4,9,100,211]:
+            x = random((n,))
+            assert_array_almost_equal(ifftshift(fftshift(x)),x)
+
+class test_fftfreq(NumpyTestCase):
+
+    def check_definition(self):
+        x = [0,1,2,3,4,-4,-3,-2,-1]
+        assert_array_almost_equal(9*fftfreq(9),x)
+        assert_array_almost_equal(9*pi*fftfreq(9,pi),x)
+        x = [0,1,2,3,4,-5,-4,-3,-2,-1]
+        assert_array_almost_equal(10*fftfreq(10),x)
+        assert_array_almost_equal(10*pi*fftfreq(10,pi),x)
+
+if __name__ == "__main__":
+    NumpyTest().run()