ref: d6352bf79ba83d3439a298ad9e10df51f6327253
dir: /python/tests/test_mfcc.py/
#! /usr/bin/env python
from _tools import parametrize, assert_raises
from numpy import random, count_nonzero
from numpy.testing import TestCase
from aubio import mfcc, cvec, float_type
buf_size = 2048
n_filters = 40
n_coeffs = 13
samplerate = 44100
new_params = ['buf_size', 'n_filters', 'n_coeffs', 'samplerate']
new_deflts = [1024, 40, 13, 44100]
class Test_aubio_mfcc(object):
members_args = 'name'
@parametrize(members_args, new_params)
def test_read_only_member(self, name):
o = mfcc()
with assert_raises((TypeError, AttributeError)):
setattr(o, name, 0)
@parametrize('name, expected', zip(new_params, new_deflts))
def test_default_param(self, name, expected):
""" test mfcc.{:s} = {:d} """.format(name, expected)
o = mfcc()
assert getattr(o, name) == expected
class aubio_mfcc_wrong_params(TestCase):
def test_wrong_buf_size(self):
with self.assertRaises(ValueError):
mfcc(buf_size = -1)
def test_wrong_n_filters(self):
with self.assertRaises(ValueError):
mfcc(n_filters = -1)
def test_wrong_n_coeffs(self):
with self.assertRaises(ValueError):
mfcc(n_coeffs = -1)
def test_wrong_samplerate(self):
with self.assertRaises(ValueError):
mfcc(samplerate = -1)
def test_wrong_input_size(self):
m = mfcc(buf_size = 1024)
with self.assertRaises(ValueError):
m(cvec(512))
class aubio_mfcc_compute(TestCase):
def test_members(self):
o = mfcc(buf_size, n_filters, n_coeffs, samplerate)
#assert_equal ([o.buf_size, o.method], [buf_size, method])
spec = cvec(buf_size)
#spec.norm[0] = 1
#spec.norm[1] = 1./2.
#print "%20s" % method, str(o(spec))
coeffs = o(spec)
self.assertEqual(coeffs.size, n_coeffs)
#print coeffs
spec.norm = random.random_sample((len(spec.norm),)).astype(float_type)
spec.phas = random.random_sample((len(spec.phas),)).astype(float_type)
#print "%20s" % method, str(o(spec))
self.assertEqual(count_nonzero(o(spec) != 0.), n_coeffs)
#print coeffs
class Test_aubio_mfcc_all_parameters(object):
run_values = [
(2048, 40, 13, 44100),
(1024, 40, 13, 44100),
(512, 40, 13, 44100),
(512, 40, 13, 16000),
(256, 40, 13, 16000),
(128, 40, 13, 16000),
(128, 40, 12, 16000),
(128, 40, 13, 15000),
(512, 40, 20, 44100),
(512, 40, 40, 44100),
(512, 40, 3, 44100),
(1024, 40, 20, 44100),
#(1024, 30, 20, 44100),
(1024, 40, 40, 44100),
(1024, 40, 3, 44100),
]
run_args = ['buf_size', 'n_filters', 'n_coeffs', 'samplerate']
@parametrize(run_args, run_values)
def test_run_with_params(self, buf_size, n_filters, n_coeffs, samplerate):
" check mfcc can run with reasonable parameters "
o = mfcc(buf_size, n_filters, n_coeffs, samplerate)
spec = cvec(buf_size)
spec.phas[0] = 0.2
for _ in range(10):
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.5)
assert m.get_scale() == 10.5
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.5)
assert m.get_power() == 2.5
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()
m(cvec(buf_size))
assert m.get_power() == 1
assert m.get_scale() == 1
if __name__ == '__main__':
from _tools import run_module_suite
run_module_suite()