source: python/tests/test_mfcc.py @ 62c2d00

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5
Last change on this file since 62c2d00 was 62c2d00, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[tests] add tests for mfcc filterbank settings

  • Property mode set to 100755
File size: 4.4 KB
RevLine 
[b6b65cb]1#! /usr/bin/env python
2
[0b6d23d]3from nose2 import main
4from nose2.tools import params
5from numpy import random, count_nonzero
6from numpy.testing import TestCase
[b6b65cb]7from aubio import mfcc, cvec, float_type
8
9buf_size = 2048
10n_filters = 40
11n_coeffs = 13
12samplerate = 44100
13
14
15new_params = ['buf_size', 'n_filters', 'n_coeffs', 'samplerate']
16new_deflts = [1024, 40, 13, 44100]
17
18class 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
41class 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
[c395732]59    def test_wrong_input_size(self):
60        m = mfcc(buf_size = 1024)
61        with self.assertRaises(ValueError):
62            m(cvec(512))
63
[b6b65cb]64class 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
85class 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
[0b6d23d]109        for _ in range(10):
110            o(spec)
[b6b65cb]111        #print coeffs
112
[62c2d00]113
114class 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.)
120        m(cvec(buf_size))
121
122    def test_set_power(self):
123        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
124        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
125        m.set_power(2.)
126        m(cvec(buf_size))
127
128    def test_set_mel_coeffs(self):
129        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
130        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
131        m.set_mel_coeffs(0., samplerate/2.)
132        m(cvec(buf_size))
133
134    def test_set_mel_coeffs_htk(self):
135        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
136        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
137        m.set_mel_coeffs_htk(0., samplerate/2.)
138        m(cvec(buf_size))
139
140    def test_set_mel_coeffs_slaney(self):
141        buf_size, n_filters, n_coeffs, samplerate = 512, 40, 10, 16000
142        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
143        m.set_mel_coeffs_slaney(samplerate)
144        m(cvec(buf_size))
145        assert m.get_power() == 1
146        assert m.get_scale() == 1
147
[b6b65cb]148if __name__ == '__main__':
[0b6d23d]149    main()
Note: See TracBrowser for help on using the repository browser.