3 from unittest import main
4 from numpy.testing import TestCase
5 from numpy.testing import assert_equal, assert_almost_equal
6 from _tools import assert_warns
10 from aubio import hztomel, meltohz
11 from aubio import hztomel_htk, meltohz_htk
14 class aubio_hztomel_test_case(TestCase):
16 def test_hztomel(self):
17 assert_equal(hztomel(0.), 0.)
18 assert_almost_equal(hztomel(400. / 3.), 2., decimal=5)
19 assert_almost_equal(hztomel(1000. / 3), 5.)
20 assert_equal(hztomel(200.), 3.)
21 assert_almost_equal(hztomel(1000.), 15)
22 assert_almost_equal(hztomel(6400), 42, decimal=5)
23 assert_almost_equal(hztomel(40960), 69, decimal=5)
25 for m in np.linspace(0, 1000, 100):
26 assert_almost_equal(hztomel(meltohz(m)) - m, 0, decimal=3)
28 def test_meltohz(self):
29 assert_equal(meltohz(0.), 0.)
30 assert_almost_equal(meltohz(2), 400. / 3., decimal=4)
31 assert_equal(meltohz(3.), 200.)
32 assert_almost_equal(meltohz(5), 1000. / 3., decimal=4)
33 assert_almost_equal(meltohz(15), 1000., decimal=4)
34 assert_almost_equal(meltohz(42), 6400., decimal=2)
35 assert_almost_equal(meltohz(69), 40960., decimal=1)
37 for f in np.linspace(0, 20000, 1000):
38 assert_almost_equal(meltohz(hztomel(f)) - f, 0, decimal=1)
40 def test_meltohz_negative(self):
41 with assert_warns(UserWarning):
42 assert_equal(meltohz(-1), 0)
44 def test_hztomel_negative(self):
45 with assert_warns(UserWarning):
46 assert_equal(hztomel(-1), 0)
49 class aubio_hztomel_htk_test_case(TestCase):
51 def test_meltohz(self):
52 assert_equal(meltohz(0, htk=True), 0)
53 assert_almost_equal(meltohz(2595, htk=True), 6300., decimal=1)
55 def test_hztomel(self):
56 assert_equal(hztomel(0, htk=True), 0)
57 assert_almost_equal(hztomel(3428.7, htk=True), 2000., decimal=1)
58 assert_almost_equal(hztomel(6300, htk=True), 2595., decimal=1)
60 def test_meltohz_negative(self):
61 with assert_warns(UserWarning):
62 assert_equal(meltohz(-1, htk=True), 0)
63 assert_almost_equal(meltohz(2000, htk=True), 3428.7, decimal=1)
64 assert_almost_equal(meltohz(1000, htk=True), 1000., decimal=1)
66 def test_hztomel_negative(self):
67 with assert_warns(UserWarning):
68 assert_equal(meltohz(-1, htk=True), 0)
69 with assert_warns(UserWarning):
70 assert_equal(hztomel(-1, htk=True), 0)
71 assert_almost_equal(hztomel(1000, htk=True), 1000., decimal=1)
73 def test_hztomel_htk(self):
74 for f in np.linspace(0, 20000, 1000):
75 assert_almost_equal(meltohz_htk(hztomel_htk(f)) - f, 0, decimal=1)
76 for f in np.linspace(0, 20000, 1000):
77 assert_almost_equal(hztomel_htk(meltohz_htk(f)) - f, 0, decimal=1)
80 class aubio_hztomel_wrong_values(TestCase):
81 """ more tests to cover all branches """
83 def test_hztomel_wrong_values(self):
84 with self.assertRaises(TypeError):
87 def test_meltohz_wrong_values(self):
88 with self.assertRaises(TypeError):
91 def test_meltohz_no_arg(self):
92 with self.assertRaises(TypeError):
95 def test_meltohz_htk_no_arg(self):
96 with self.assertRaises(TypeError):
99 def test_hztomel_htk_wrong_values(self):
100 with self.assertRaises(TypeError):
103 def test_hztomel_htk_false(self):
104 assert hztomel(120, htk=False) == hztomel(120)
106 def test_meltohz_htk_false(self):
107 assert meltohz(12, htk=False) == meltohz(12)
110 if __name__ == '__main__':
111 from unittest import main