[tests] mfcc norm/scale with real values
[aubio.git] / python / tests / test_mfcc.py
1 #! /usr/bin/env python
2
3 from nose2 import main
4 from nose2.tools import params
5 from numpy import random, count_nonzero
6 from numpy.testing import TestCase
7 from aubio import mfcc, cvec, float_type
8
9 buf_size = 2048
10 n_filters = 40
11 n_coeffs = 13
12 samplerate = 44100
13
14
15 new_params = ['buf_size', 'n_filters', 'n_coeffs', 'samplerate']
16 new_deflts = [1024, 40, 13, 44100]
17
18 class aubio_mfcc(TestCase):
19
20     def setUp(self):
21         self.o = mfcc()
22
23     def test_default_creation(self):
24         pass
25
26     def test_delete(self):
27         del self.o
28
29     @params(*new_params)
30     def test_read_only_member(self, name):
31         o = self.o
32         with self.assertRaises((TypeError, AttributeError)):
33             setattr(o, name, 0)
34
35     @params(*zip(new_params, new_deflts))
36     def test_default_param(self, name, expected):
37         """ test mfcc.{:s} = {:d} """.format(name, expected)
38         o = self.o
39         self.assertEqual( getattr(o, name), expected)
40
41 class aubio_mfcc_wrong_params(TestCase):
42
43     def test_wrong_buf_size(self):
44         with self.assertRaises(ValueError):
45             mfcc(buf_size = -1)
46
47     def test_wrong_n_filters(self):
48         with self.assertRaises(ValueError):
49             mfcc(n_filters = -1)
50
51     def test_wrong_n_coeffs(self):
52         with self.assertRaises(ValueError):
53             mfcc(n_coeffs = -1)
54
55     def test_wrong_samplerate(self):
56         with self.assertRaises(ValueError):
57             mfcc(samplerate = -1)
58
59     def test_wrong_input_size(self):
60         m = mfcc(buf_size = 1024)
61         with self.assertRaises(ValueError):
62             m(cvec(512))
63
64 class aubio_mfcc_compute(TestCase):
65
66     def test_members(self):
67
68         o = mfcc(buf_size, n_filters, n_coeffs, samplerate)
69         #assert_equal ([o.buf_size, o.method], [buf_size, method])
70
71         spec = cvec(buf_size)
72         #spec.norm[0] = 1
73         #spec.norm[1] = 1./2.
74         #print "%20s" % method, str(o(spec))
75         coeffs = o(spec)
76         self.assertEqual(coeffs.size, n_coeffs)
77         #print coeffs
78         spec.norm = random.random_sample((len(spec.norm),)).astype(float_type)
79         spec.phas = random.random_sample((len(spec.phas),)).astype(float_type)
80         #print "%20s" % method, str(o(spec))
81         self.assertEqual(count_nonzero(o(spec) != 0.), n_coeffs)
82         #print coeffs
83
84
85 class aubio_mfcc_all_parameters(TestCase):
86
87     @params(
88             (2048, 40, 13, 44100),
89             (1024, 40, 13, 44100),
90             (512, 40, 13, 44100),
91             (512, 40, 13, 16000),
92             (256, 40, 13, 16000),
93             (128, 40, 13, 16000),
94             (128, 40, 12, 16000),
95             (128, 40, 13, 15000),
96             (512, 40, 20, 44100),
97             (512, 40, 40, 44100),
98             (512, 40, 3, 44100),
99             (1024, 40, 20, 44100),
100             #(1024, 30, 20, 44100),
101             (1024, 40, 40, 44100),
102             (1024, 40, 3, 44100),
103             )
104     def test_run_with_params(self, buf_size, n_filters, n_coeffs, samplerate):
105         " check mfcc can run with reasonable parameters "
106         o = mfcc(buf_size, n_filters, n_coeffs, samplerate)
107         spec = cvec(buf_size)
108         spec.phas[0] = 0.2
109         for _ in range(10):
110             o(spec)
111         #print coeffs
112
113
114 class aubio_mfcc_fb_params(TestCase):
115
116     def test_set_scale(self):
117         buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
118         m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
119         m.set_scale(10.5)
120         assert m.get_scale() == 10.5
121         m(cvec(buf_size))
122
123     def test_set_power(self):
124         buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
125         m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
126         m.set_power(2.5)
127         assert m.get_power() == 2.5
128         m(cvec(buf_size))
129
130     def test_set_mel_coeffs(self):
131         buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
132         m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
133         m.set_mel_coeffs(0., samplerate/2.)
134         m(cvec(buf_size))
135
136     def test_set_mel_coeffs_htk(self):
137         buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
138         m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
139         m.set_mel_coeffs_htk(0., samplerate/2.)
140         m(cvec(buf_size))
141
142     def test_set_mel_coeffs_slaney(self):
143         buf_size, n_filters, n_coeffs, samplerate = 512, 40, 10, 16000
144         m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
145         m.set_mel_coeffs_slaney()
146         m(cvec(buf_size))
147         assert m.get_power() == 1
148         assert m.get_scale() == 1
149
150 if __name__ == '__main__':
151     main()