#! /usr/bin/env python from _tools import parametrize, assert_raisesfrom numpy import random, count_nonzerofrom numpy.testing import TestCasefrom aubio import mfcc, cvec, float_type buf_size = 2048n_filters = 40n_coeffs = 13samplerate = 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()