source: python/tests/test_fft.py @ 376d5e9

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since 376d5e9 was 0a42239, checked in by Paul Brossier <piem@piem.org>, 11 years ago

python/tests/test_fft.py: update, indent

  • Property mode set to 100755
File size: 3.8 KB
Line 
1#! /usr/bin/env python
2
3from numpy.testing import TestCase, run_module_suite
4from numpy.testing import assert_equal, assert_almost_equal
5from aubio import fvec, fft, cvec
6from numpy import array, shape
7from math import pi
8
9class aubio_fft_test_case(TestCase):
10
11    def test_members(self):
12        """ check members are set correctly """
13        win_s = 2048
14        f = fft(win_s)
15        assert_equal (f.win_s, win_s)
16
17    def test_output_dimensions(self):
18        """ check the dimensions of output """
19        win_s = 1024
20        timegrain = fvec(win_s)
21        f = fft (win_s)
22        fftgrain = f (timegrain)
23        assert_equal (shape(fftgrain.norm), (win_s/2+1,))
24        assert_equal (shape(fftgrain.phas), (win_s/2+1,))
25
26    def test_zeros(self):
27        """ check the transform of zeros is all zeros """
28        win_s = 512
29        timegrain = fvec(win_s)
30        f = fft (win_s)
31        fftgrain = f (timegrain)
32        assert_equal ( fftgrain.norm, 0 )
33        assert_equal ( fftgrain.phas, 0 )
34
35    def test_impulse(self):
36        """ check the transform of one impulse at a random place """
37        from random import random
38        from math import floor
39        win_s = 256
40        i = floor(random()*win_s)
41        impulse = pi * random()
42        f = fft(win_s)
43        timegrain = fvec(win_s)
44        timegrain[i] = impulse
45        fftgrain = f ( timegrain )
46        #self.plot_this ( fftgrain.phas )
47        assert_almost_equal ( fftgrain.norm, impulse, decimal = 6 )
48        assert_equal ( fftgrain.phas <= pi, True)
49        assert_equal ( fftgrain.phas >= -pi, True)
50
51    def test_impulse_negative(self):
52        """ check the transform of one impulse at a random place """
53        from random import random
54        from math import floor
55        win_s = 256
56        i = 0
57        impulse = -10.
58        f = fft(win_s)
59        timegrain = fvec(win_s)
60        timegrain[i] = impulse
61        fftgrain = f ( timegrain )
62        #self.plot_this ( fftgrain.phas )
63        assert_almost_equal ( fftgrain.norm, abs(impulse), decimal = 6 )
64        if impulse < 0:
65            # phase can be pi or -pi, as it is not unwrapped
66            assert_almost_equal ( abs(fftgrain.phas[1:-1]) , pi, decimal = 6 )
67            assert_almost_equal ( fftgrain.phas[0], pi, decimal = 6)
68            assert_almost_equal ( fftgrain.phas[-1], pi, decimal = 6)
69        else:
70            assert_equal ( fftgrain.phas[1:-1] == 0, True)
71            assert_equal ( fftgrain.phas[0] == 0, True)
72            assert_equal ( fftgrain.phas[-1] == 0, True)
73        # now check the resynthesis
74        synthgrain = f.rdo ( fftgrain )
75        #self.plot_this ( fftgrain.phas.T )
76        assert_equal ( fftgrain.phas <= pi, True)
77        assert_equal ( fftgrain.phas >= -pi, True)
78        #self.plot_this ( synthgrain - timegrain )
79        assert_almost_equal ( synthgrain, timegrain, decimal = 6 )
80
81    def test_impulse_at_zero(self):
82        """ check the transform of one impulse at a index 0 """
83        win_s = 1024
84        impulse = pi
85        f = fft(win_s)
86        timegrain = fvec(win_s)
87        timegrain[0] = impulse
88        fftgrain = f ( timegrain )
89        #self.plot_this ( fftgrain.phas )
90        assert_equal ( fftgrain.phas[0], 0)
91        # could be 0 or -0 depending on fft implementation (0 for fftw3, -0 for ooura)
92        assert_almost_equal ( fftgrain.phas[1], 0)
93        assert_almost_equal ( fftgrain.norm[0], impulse, decimal = 6 )
94
95    def test_rdo_before_do(self):
96        """ check running fft.rdo before fft.do works """
97        win_s = 1024
98        impulse = pi
99        f = fft(win_s)
100        fftgrain = cvec(win_s)
101        t = f.rdo( fftgrain )
102        assert_equal ( t, 0 )
103
104    def plot_this(self, this):
105        from pylab import plot, show
106        plot ( this )
107        show ()
108
109if __name__ == '__main__':
110    from unittest import main
111    main()
112
Note: See TracBrowser for help on using the repository browser.