From: Paul Brossier <piem@piem.org>
Date: Sat, 17 Nov 2018 18:07:46 +0000 (+0100)
Subject: [tests] add tests for mfcc filterbank settings
X-Git-Tag: 0.4.8~30
X-Git-Url: https://git.aubio.org/?a=commitdiff_plain;h=62c2d005cd49b3d20341621fe70b3de1087ab632;p=aubio.git

[tests] add tests for mfcc filterbank settings
---

diff --git a/python/tests/test_mfcc.py b/python/tests/test_mfcc.py
index e7f3b186..45b5ce40 100755
--- a/python/tests/test_mfcc.py
+++ b/python/tests/test_mfcc.py
@@ -110,5 +110,40 @@ class aubio_mfcc_all_parameters(TestCase):
             o(spec)
         #print coeffs
 
+
+class aubio_mfcc_fb_params(TestCase):
+
+    def test_set_scale(self):
+        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
+        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
+        m.set_scale(10.)
+        m(cvec(buf_size))
+
+    def test_set_power(self):
+        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
+        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
+        m.set_power(2.)
+        m(cvec(buf_size))
+
+    def test_set_mel_coeffs(self):
+        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
+        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
+        m.set_mel_coeffs(0., samplerate/2.)
+        m(cvec(buf_size))
+
+    def test_set_mel_coeffs_htk(self):
+        buf_size, n_filters, n_coeffs, samplerate = 512, 20, 10, 16000
+        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
+        m.set_mel_coeffs_htk(0., samplerate/2.)
+        m(cvec(buf_size))
+
+    def test_set_mel_coeffs_slaney(self):
+        buf_size, n_filters, n_coeffs, samplerate = 512, 40, 10, 16000
+        m = mfcc(buf_size, n_filters, n_coeffs, samplerate)
+        m.set_mel_coeffs_slaney(samplerate)
+        m(cvec(buf_size))
+        assert m.get_power() == 1
+        assert m.get_scale() == 1
+
 if __name__ == '__main__':
     main()