Changeset e116e19 for python/tests/test_filterbank.py

Timestamp:

Sep 14, 2017, 4:33:21 PM (7 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master

Children:

9a52962

Parents:

c4fc0f2

Message:

python/tests/test_filterbank*.py: clean-up, improve get_coeff tests

File:

• python/tests/test_filterbank.py (modified) (3 diffs)

python/tests/test_filterbank.py

@@ -1 +1 @@
 #! /usr/bin/env python
 
-from unittest import main
-from numpy.testing import TestCase
-from numpy.testing import assert_equal, assert_almost_equal
 import numpy as np
+from numpy.testing import TestCase, assert_equal, assert_almost_equal
+
 from aubio import cvec, filterbank, float_type
 from .utils import array_from_text_file
@@ -63 +62 @@
         assert_almost_equal ( expected, f.get_coeffs() )
 
+    def test_mfcc_coeffs_get_coeffs(self):
+        f = filterbank(40, 512)
+        coeffs = f.get_coeffs()
+        self.assertIsInstance(coeffs, np.ndarray)
+        assert_equal (coeffs, 0)
+        assert_equal (np.shape(coeffs), (40, 512 / 2 + 1))
+
 class aubio_filterbank_wrong_values(TestCase):
 
@@ -82 +88 @@
 
 if __name__ == '__main__':
-    main()
+    import nose2
+    nose2.main()