[398c1c5] | 1 | import unittest |
---|
| 2 | import math |
---|
| 3 | |
---|
| 4 | from aubio.aubiowrapper import * |
---|
| 5 | |
---|
[c721874] | 6 | buf_size = 8092 |
---|
| 7 | channels = 4 |
---|
| 8 | |
---|
| 9 | precision = 6 |
---|
[398c1c5] | 10 | |
---|
| 11 | class aubio_mfft_test_case(unittest.TestCase): |
---|
| 12 | |
---|
| 13 | def setUp(self): |
---|
| 14 | self.o = new_aubio_mfft(buf_size, channels) |
---|
| 15 | |
---|
| 16 | def tearDown(self): |
---|
| 17 | del_aubio_mfft(self.o) |
---|
| 18 | |
---|
[2d8880d5] | 19 | def test_create(self): |
---|
| 20 | """ test creation and deletion of fft object """ |
---|
[398c1c5] | 21 | pass |
---|
| 22 | |
---|
| 23 | def test_aubio_mfft_do_zeroes(self): |
---|
| 24 | """ test aubio_mfft_do on zeroes """ |
---|
| 25 | input = new_fvec(buf_size, channels) |
---|
| 26 | fftgrain = new_cvec(buf_size, channels) |
---|
| 27 | for index in range(buf_size): |
---|
| 28 | for channel in range(channels): |
---|
| 29 | self.assertEqual(0., fvec_read_sample(input, channel, index)) |
---|
| 30 | aubio_mfft_do(self.o, input, fftgrain) |
---|
| 31 | for index in range(buf_size/2+1): |
---|
| 32 | for channel in range(channels): |
---|
| 33 | self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) |
---|
| 34 | for index in range(buf_size/2+1): |
---|
| 35 | for channel in range(channels): |
---|
| 36 | self.assertEqual(0., cvec_read_phas(fftgrain, channel, index)) |
---|
| 37 | del fftgrain |
---|
| 38 | del input |
---|
| 39 | |
---|
| 40 | def test_aubio_mfft_rdo_zeroes(self): |
---|
| 41 | """ test aubio_mfft_rdo on zeroes """ |
---|
| 42 | fftgrain = new_cvec(buf_size, channels) |
---|
| 43 | output = new_fvec(buf_size, channels) |
---|
| 44 | aubio_mfft_rdo(self.o, fftgrain, output) |
---|
| 45 | # check output |
---|
| 46 | for index in range(buf_size): |
---|
| 47 | for channel in range(channels): |
---|
| 48 | self.assertEqual(0., fvec_read_sample(output, channel, index)) |
---|
| 49 | del fftgrain |
---|
| 50 | del output |
---|
| 51 | |
---|
| 52 | def test_aubio_mfft_do_impulse(self): |
---|
[c721874] | 53 | """ test aubio_mfft_do with an impulse on one channel """ |
---|
[398c1c5] | 54 | input = new_fvec(buf_size, channels) |
---|
| 55 | fftgrain = new_cvec(buf_size, channels) |
---|
| 56 | # write impulse in channel 0, sample 0. |
---|
[c721874] | 57 | some_constant = 0.3412432456 |
---|
| 58 | fvec_write_sample(input, some_constant, 0, 0) |
---|
[398c1c5] | 59 | aubio_mfft_do(self.o, input, fftgrain) |
---|
| 60 | # check norm |
---|
| 61 | for index in range(buf_size/2+1): |
---|
[c721874] | 62 | self.assertAlmostEqual(some_constant, cvec_read_norm(fftgrain, 0, index), precision) |
---|
[398c1c5] | 63 | for index in range(buf_size/2+1): |
---|
| 64 | for channel in range(1, channels): |
---|
| 65 | self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) |
---|
| 66 | # check phas |
---|
| 67 | for index in range(buf_size/2+1): |
---|
| 68 | for channel in range(channels): |
---|
| 69 | self.assertEqual(0., cvec_read_phas(fftgrain, channel, index)) |
---|
| 70 | del fftgrain |
---|
| 71 | del input |
---|
| 72 | |
---|
[c721874] | 73 | def test_aubio_mfft_do_constant(self): |
---|
| 74 | """ test aubio_mfft_do with a constant on one channel """ |
---|
| 75 | input = new_fvec(buf_size, channels) |
---|
| 76 | fftgrain = new_cvec(buf_size, channels) |
---|
| 77 | # write impulse in channel 0, sample 0. |
---|
| 78 | some_constant = 0.003412432456 |
---|
| 79 | for index in range(1,buf_size): |
---|
| 80 | fvec_write_sample(input, some_constant, 0, index) |
---|
| 81 | aubio_mfft_do(self.o, input, fftgrain) |
---|
| 82 | # check norm and phase == 0 in all other channels |
---|
| 83 | for index in range(buf_size/2+1): |
---|
| 84 | for channel in range(1, channels): |
---|
| 85 | self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) |
---|
| 86 | # check norm and phase == 0 in first first and last bin of first channel |
---|
| 87 | self.assertAlmostEqual((buf_size-1)*some_constant, cvec_read_norm(fftgrain, 0, 0), precision) |
---|
| 88 | self.assertEqual(0., cvec_read_phas(fftgrain, 0, 0)) |
---|
| 89 | self.assertEqual(0., cvec_read_norm(fftgrain, 0, buf_size/2+1)) |
---|
| 90 | self.assertEqual(0., cvec_read_phas(fftgrain, 0, buf_size/2+1)) |
---|
| 91 | # check unwrap2pi(phas) ~= pi everywhere but in first bin |
---|
| 92 | for index in range(1,buf_size/2+1): |
---|
| 93 | self.assertAlmostEqual ( math.pi, aubio_unwrap2pi(cvec_read_phas(fftgrain, 0, index)), precision) |
---|
| 94 | self.assertAlmostEqual(some_constant, cvec_read_norm(fftgrain, 0, index), precision) |
---|
| 95 | del fftgrain |
---|
| 96 | del input |
---|
| 97 | |
---|
[398c1c5] | 98 | def test_aubio_mfft_do_impulse_multichannel(self): |
---|
| 99 | " test aubio_mfft_do on impulse two channels " |
---|
| 100 | input = new_fvec(buf_size, channels) |
---|
| 101 | fftgrain = new_cvec(buf_size, channels) |
---|
| 102 | # put an impulse in first an last channel, at first and last index |
---|
| 103 | fvec_write_sample(input, 1., 0, 0) |
---|
| 104 | fvec_write_sample(input, 1., channels-1, 0) |
---|
| 105 | aubio_mfft_do(self.o, input, fftgrain) |
---|
| 106 | # check the norm |
---|
| 107 | for index in range(buf_size/2+1): |
---|
| 108 | self.assertEqual(1., cvec_read_norm(fftgrain, 0, index)) |
---|
| 109 | for index in range(buf_size/2+1): |
---|
| 110 | for channel in range(1, channels-1): |
---|
| 111 | self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) |
---|
| 112 | for index in range(buf_size/2+1): |
---|
| 113 | self.assertEqual(1., cvec_read_norm(fftgrain, channels-1, index)) |
---|
| 114 | # check the phase |
---|
| 115 | for index in range(buf_size/2+1): |
---|
| 116 | for channel in range(channels): |
---|
| 117 | self.assertEqual(0., cvec_read_phas(fftgrain, channel, index)) |
---|
| 118 | del fftgrain |
---|
| 119 | del input |
---|
| 120 | |
---|
| 121 | def test_aubio_mfft_rdo_impulse(self): |
---|
| 122 | """ test aubio_mfft_rdo on impulse """ |
---|
| 123 | fftgrain = new_cvec(buf_size, channels) |
---|
[c721874] | 124 | for channel in range(channels): |
---|
| 125 | cvec_write_norm(fftgrain, 1., channel, 0) |
---|
[398c1c5] | 126 | output = new_fvec(buf_size, channels) |
---|
| 127 | aubio_mfft_rdo(self.o, fftgrain, output) |
---|
| 128 | for index in range(buf_size/2+1): |
---|
| 129 | for channel in range(channels): |
---|
[c721874] | 130 | self.assertAlmostEqual(fvec_read_sample(output, channel, index), 1./buf_size, precision) |
---|
[398c1c5] | 131 | del fftgrain |
---|
| 132 | del output |
---|
| 133 | |
---|
[2d8880d5] | 134 | def test_aubio_mfft_do_back_and_forth(self): |
---|
| 135 | """ test aubio_mfft_rdo on a constant """ |
---|
| 136 | input = new_fvec(buf_size, channels) |
---|
| 137 | output = new_fvec(buf_size, channels) |
---|
| 138 | fftgrain = new_cvec(buf_size, channels) |
---|
| 139 | for index in range(buf_size/2+1): |
---|
| 140 | for channel in range(channels): |
---|
| 141 | fvec_write_sample(input, 0.67, channel, index) |
---|
| 142 | aubio_mfft_do(self.o, input, fftgrain) |
---|
| 143 | aubio_mfft_rdo(self.o, fftgrain, output) |
---|
| 144 | for index in range(buf_size/2+1): |
---|
| 145 | for channel in range(channels): |
---|
[c721874] | 146 | self.assertAlmostEqual(fvec_read_sample(output, channel, index), 0.67, precision) |
---|
[2d8880d5] | 147 | del fftgrain |
---|
| 148 | del output |
---|
| 149 | |
---|
[398c1c5] | 150 | if __name__ == '__main__': unittest.main() |
---|