| |
|
|
| __usage__ = """ |
| Build fftpack: |
| python setup_fftpack.py build |
| Run tests if scipy is installed: |
| python -c 'import scipy;scipy.fftpack.test(<level>)' |
| Run tests if fftpack is not installed: |
| python tests/test_helper.py [<level>] |
| """ |
|
|
| from numpy.testing import assert_array_almost_equal |
| from scipy.fftpack import fftshift, ifftshift, fftfreq, rfftfreq |
|
|
| from numpy import pi, random |
|
|
| class TestFFTShift: |
|
|
| def test_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 test_inverse(self): |
| for n in [1,4,9,100,211]: |
| x = random.random((n,)) |
| assert_array_almost_equal(ifftshift(fftshift(x)),x) |
|
|
|
|
| class TestFFTFreq: |
|
|
| def test_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) |
|
|
|
|
| class TestRFFTFreq: |
|
|
| def test_definition(self): |
| x = [0,1,1,2,2,3,3,4,4] |
| assert_array_almost_equal(9*rfftfreq(9),x) |
| assert_array_almost_equal(9*pi*rfftfreq(9,pi),x) |
| x = [0,1,1,2,2,3,3,4,4,5] |
| assert_array_almost_equal(10*rfftfreq(10),x) |
| assert_array_almost_equal(10*pi*rfftfreq(10,pi),x) |
|
|
