summaryrefslogtreecommitdiff
path: root/numpy/fft/tests/test_fftpack.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/fft/tests/test_fftpack.py')
-rw-r--r--numpy/fft/tests/test_fftpack.py14
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()