summaryrefslogtreecommitdiff
path: root/numpy/fft/tests/test_fftpack.py
diff options
context:
space:
mode:
authorJarrod Millman <millman@berkeley.edu>2007-12-29 01:32:27 +0000
committerJarrod Millman <millman@berkeley.edu>2007-12-29 01:32:27 +0000
commite7191fc96a676b5ea444f23501a3db61c2be602e (patch)
treefac59ea48855eb4b0a926303e4f898e5f5643bfa /numpy/fft/tests/test_fftpack.py
parente31192fd7799a6456459d0e6142ddef8d46b8d08 (diff)
downloadnumpy-e7191fc96a676b5ea444f23501a3db61c2be602e.tar.gz
janitorial work
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()