source: python/tests/test_fft.py @ 01d4d19

feature/autosinkfeature/constantqfeature/pitchshiftfeature/pydocstringsfeature/timestretch
Last change on this file since 01d4d19 was 01d4d19, checked in by Paul Brossier <piem@piem.org>, 23 months ago

Merge branch 'fix/oddfft' (closes #207)

  • Property mode set to 100755
File size: 7.0 KB
RevLine 
[75e715f]1#! /usr/bin/env python
[1ebf8770]2
[0b6d23d]3from unittest import main
[ad5eae8]4from numpy.testing import TestCase
[0536612]5from numpy.testing import assert_equal, assert_almost_equal
[0b6d23d]6import numpy as np
[4c01c0f]7from aubio import fvec, fft, cvec
[0b6d23d]8from math import pi, floor
9from random import random
[0536612]10
11class aubio_fft_test_case(TestCase):
12
[0a42239]13    def test_members(self):
14        """ check members are set correctly """
15        win_s = 2048
16        f = fft(win_s)
17        assert_equal (f.win_s, win_s)
[0536612]18
[0a42239]19    def test_output_dimensions(self):
20        """ check the dimensions of output """
21        win_s = 1024
22        timegrain = fvec(win_s)
23        f = fft (win_s)
24        fftgrain = f (timegrain)
[ad5eae8]25        del f
26        assert_equal (fftgrain.norm.shape, (win_s/2+1,))
27        assert_equal (fftgrain.phas.shape, (win_s/2+1,))
[0536612]28
[0a42239]29    def test_zeros(self):
30        """ check the transform of zeros is all zeros """
31        win_s = 512
32        timegrain = fvec(win_s)
33        f = fft (win_s)
34        fftgrain = f (timegrain)
35        assert_equal ( fftgrain.norm, 0 )
[d98f3c0]36        try:
37            assert_equal ( fftgrain.phas, 0 )
38        except AssertionError:
39            assert_equal (fftgrain.phas[fftgrain.phas > 0], +pi)
40            assert_equal (fftgrain.phas[fftgrain.phas < 0], -pi)
41            assert_equal (np.abs(fftgrain.phas[np.abs(fftgrain.phas) != pi]), 0)
42            self.skipTest('fft(fvec(%d)).phas != +0, ' % win_s \
43                    + 'This is expected when using fftw3 on powerpc.')
[0536612]44
[0a42239]45    def test_impulse(self):
46        """ check the transform of one impulse at a random place """
47        win_s = 256
[2f4d9b7]48        i = int(floor(random()*win_s))
[0a42239]49        impulse = pi * random()
50        f = fft(win_s)
51        timegrain = fvec(win_s)
52        timegrain[i] = impulse
53        fftgrain = f ( timegrain )
54        #self.plot_this ( fftgrain.phas )
55        assert_almost_equal ( fftgrain.norm, impulse, decimal = 6 )
56        assert_equal ( fftgrain.phas <= pi, True)
57        assert_equal ( fftgrain.phas >= -pi, True)
[0536612]58
[0a42239]59    def test_impulse_negative(self):
[0b6d23d]60        """ check the transform of a negative impulse at a random place """
[0a42239]61        win_s = 256
[0b6d23d]62        i = int(floor(random()*win_s))
63        impulse = -.1
[0a42239]64        f = fft(win_s)
65        timegrain = fvec(win_s)
[0b6d23d]66        timegrain[0] = 0
[0a42239]67        timegrain[i] = impulse
68        fftgrain = f ( timegrain )
69        #self.plot_this ( fftgrain.phas )
[0b6d23d]70        assert_almost_equal ( fftgrain.norm, abs(impulse), decimal = 5 )
[0a42239]71        if impulse < 0:
72            # phase can be pi or -pi, as it is not unwrapped
[0b6d23d]73            #assert_almost_equal ( abs(fftgrain.phas[1:-1]) , pi, decimal = 6 )
[0a42239]74            assert_almost_equal ( fftgrain.phas[0], pi, decimal = 6)
[0b6d23d]75            assert_almost_equal ( np.fmod(fftgrain.phas[-1], pi), 0, decimal = 6)
[0a42239]76        else:
[0b6d23d]77            #assert_equal ( fftgrain.phas[1:-1] == 0, True)
78            assert_equal ( fftgrain.phas[0], 0)
79            assert_almost_equal ( np.fmod(fftgrain.phas[-1], pi), 0, decimal = 6)
[0a42239]80        # now check the resynthesis
81        synthgrain = f.rdo ( fftgrain )
82        #self.plot_this ( fftgrain.phas.T )
83        assert_equal ( fftgrain.phas <= pi, True)
84        assert_equal ( fftgrain.phas >= -pi, True)
85        #self.plot_this ( synthgrain - timegrain )
86        assert_almost_equal ( synthgrain, timegrain, decimal = 6 )
[0536612]87
[0a42239]88    def test_impulse_at_zero(self):
89        """ check the transform of one impulse at a index 0 """
90        win_s = 1024
91        impulse = pi
92        f = fft(win_s)
93        timegrain = fvec(win_s)
94        timegrain[0] = impulse
95        fftgrain = f ( timegrain )
96        #self.plot_this ( fftgrain.phas )
97        assert_equal ( fftgrain.phas[0], 0)
98        # could be 0 or -0 depending on fft implementation (0 for fftw3, -0 for ooura)
99        assert_almost_equal ( fftgrain.phas[1], 0)
100        assert_almost_equal ( fftgrain.norm[0], impulse, decimal = 6 )
[0536612]101
[0a42239]102    def test_rdo_before_do(self):
103        """ check running fft.rdo before fft.do works """
104        win_s = 1024
105        f = fft(win_s)
106        fftgrain = cvec(win_s)
107        t = f.rdo( fftgrain )
108        assert_equal ( t, 0 )
[0536612]109
[0a42239]110    def plot_this(self, this):
111        from pylab import plot, show
112        plot ( this )
113        show ()
[0536612]114
[ee092a8]115    def test_local_fftgrain(self):
116        """ check aubio.fft() result can be accessed after deletion """
117        def compute_grain(impulse):
118            win_s = 1024
119            timegrain = fvec(win_s)
120            timegrain[0] = impulse
121            f = fft(win_s)
122            fftgrain = f ( timegrain )
123            return fftgrain
124        impulse = pi
125        fftgrain = compute_grain(impulse)
126        assert_equal ( fftgrain.phas[0], 0)
127        assert_almost_equal ( fftgrain.phas[1], 0)
128        assert_almost_equal ( fftgrain.norm[0], impulse, decimal = 6 )
129
130    def test_local_reconstruct(self):
131        """ check aubio.fft.rdo() result can be accessed after deletion """
132        def compute_grain(impulse):
133            win_s = 1024
134            timegrain = fvec(win_s)
135            timegrain[0] = impulse
136            f = fft(win_s)
137            fftgrain = f ( timegrain )
138            r = f.rdo(fftgrain)
139            return r
140        impulse = pi
141        r = compute_grain(impulse)
142        assert_almost_equal ( r[0], impulse, decimal = 6)
143        assert_almost_equal ( r[1:], 0)
144
[01d4d19]145class aubio_fft_odd_sizes(TestCase):
146
147    def test_reconstruct_with_odd_size(self):
148        win_s = 29
149        self.recontruct(win_s, 'odd sizes not supported')
150
151    def test_reconstruct_with_radix15(self):
152        win_s = 2 ** 4 * 15
153        self.recontruct(win_s, 'radix 15 supported')
154
155    def test_reconstruct_with_radix5(self):
156        win_s = 2 ** 4 * 5
157        self.recontruct(win_s, 'radix 5 supported')
158
159    def test_reconstruct_with_radix3(self):
160        win_s = 2 ** 4 * 3
161        self.recontruct(win_s, 'radix 3 supported')
162
163    def recontruct(self, win_s, skipMessage):
164        try:
165            f = fft(win_s)
166        except RuntimeError:
167            self.skipTest(skipMessage)
168        input_signal = fvec(win_s)
169        input_signal[win_s//2] = 1
170        c = f(input_signal)
171        output_signal = f.rdo(c)
172        assert_almost_equal(input_signal, output_signal)
173
174class aubio_fft_wrong_params(TestCase):
175
[d03ee4b]176    def test_large_input_timegrain(self):
177        win_s = 1024
178        f = fft(win_s)
179        t = fvec(win_s + 1)
180        with self.assertRaises(ValueError):
[437ef07]181            f(t)
[d03ee4b]182
183    def test_small_input_timegrain(self):
184        win_s = 1024
185        f = fft(win_s)
186        t = fvec(1)
187        with self.assertRaises(ValueError):
188            f(t)
189
[437ef07]190    def test_large_input_fftgrain(self):
191        win_s = 1024
192        f = fft(win_s)
[7498e48]193        s = cvec(win_s + 5)
[437ef07]194        with self.assertRaises(ValueError):
195            f.rdo(s)
196
[ad5eae8]197    def test_small_input_fftgrain(self):
[437ef07]198        win_s = 1024
199        f = fft(win_s)
200        s = cvec(16)
201        with self.assertRaises(ValueError):
202            f.rdo(s)
203
[fcef3fd]204    def test_wrong_buf_size(self):
205        win_s = -1
206        with self.assertRaises(ValueError):
207            fft(win_s)
208
209    def test_buf_size_too_small(self):
210        win_s = 1
211        with self.assertRaises(RuntimeError):
212            fft(win_s)
213
[0536612]214if __name__ == '__main__':
[0a42239]215    main()
Note: See TracBrowser for help on using the repository browser.