Changeset f264b17 for python/tests/test_filter.py

Timestamp:

Jun 22, 2016, 1:00:10 PM (8 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, pitchshift, sampler, timestretch, yinfft+

Children:

4b9443c4

Parents:

60fc05b (diff), 6769586 (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

Merge branch 'master' into notes

File:

• python/tests/test_filter.py (modified) (1 diff)

python/tests/test_filter.py

@@ -1 +1 @@
 #! /usr/bin/env python
 
+from unittest import main
 from numpy.testing import TestCase, assert_equal, assert_almost_equal
 from aubio import fvec, digital_filter
-from numpy import array
 from utils import array_from_text_file
 
 class aubio_filter_test_case(TestCase):
 
-  def test_members(self):
-    f = digital_filter()
-    assert_equal (f.order, 7)
-    f = digital_filter(5)
-    assert_equal (f.order, 5)
-    f(fvec())
-  
-  def test_cweighting_error(self):
-    f = digital_filter (2)
-    self.assertRaises ( ValueError, f.set_c_weighting, 44100 )
-    f = digital_filter (8)
-    self.assertRaises ( ValueError, f.set_c_weighting, 44100 )
-    f = digital_filter (5)
-    self.assertRaises ( ValueError, f.set_c_weighting, 4000 )
-    f = digital_filter (5)
-    self.assertRaises ( ValueError, f.set_c_weighting, 193000 )
-    f = digital_filter (7)
-    self.assertRaises ( ValueError, f.set_a_weighting, 193000 )
-    f = digital_filter (5)
-    self.assertRaises ( ValueError, f.set_a_weighting, 192000 )
+    def test_members(self):
+        f = digital_filter()
+        assert_equal (f.order, 7)
+        f = digital_filter(5)
+        assert_equal (f.order, 5)
+        f(fvec())
 
-  def test_c_weighting(self):
-    expected = array_from_text_file('c_weighting_test_simple.expected')
-    f = digital_filter(5)
-    f.set_c_weighting(44100)
-    v = fvec(32)
-    v[12] = .5
-    u = f(v)
-    assert_almost_equal (expected[1], u)
+    def test_cweighting_error(self):
+        f = digital_filter (2)
+        self.assertRaises ( ValueError, f.set_c_weighting, 44100 )
+        f = digital_filter (8)
+        self.assertRaises ( ValueError, f.set_c_weighting, 44100 )
+        f = digital_filter (5)
+        self.assertRaises ( ValueError, f.set_c_weighting, 4000 )
+        f = digital_filter (5)
+        self.assertRaises ( ValueError, f.set_c_weighting, 193000 )
+        f = digital_filter (7)
+        self.assertRaises ( ValueError, f.set_a_weighting, 193000 )
+        f = digital_filter (5)
+        self.assertRaises ( ValueError, f.set_a_weighting, 192000 )
 
-  def test_c_weighting_8000(self):
-    expected = array_from_text_file('c_weighting_test_simple_8000.expected')
-    f = digital_filter(5)
-    f.set_c_weighting(8000)
-    v = fvec(32)
-    v[12] = .5
-    u = f(v)
-    assert_almost_equal (expected[1], u)
+    def test_c_weighting(self):
+        expected = array_from_text_file('c_weighting_test_simple.expected')
+        f = digital_filter(5)
+        f.set_c_weighting(44100)
+        v = fvec(32)
+        v[12] = .5
+        u = f(v)
+        assert_almost_equal (expected[1], u)
 
-  def test_a_weighting(self):
-    expected = array_from_text_file('a_weighting_test_simple.expected')
-    f = digital_filter(7)
-    f.set_a_weighting(44100)
-    v = fvec(32)
-    v[12] = .5
-    u = f(v)
-    assert_almost_equal (expected[1], u)
+    def test_c_weighting_8000(self):
+        expected = array_from_text_file('c_weighting_test_simple_8000.expected')
+        f = digital_filter(5)
+        f.set_c_weighting(8000)
+        v = fvec(32)
+        v[12] = .5
+        u = f(v)
+        assert_almost_equal (expected[1], u)
 
-  def test_a_weighting_parted(self):
-    expected = array_from_text_file('a_weighting_test_simple.expected')
-    f = digital_filter(7)
-    f.set_a_weighting(44100)
-    v = fvec(16)
-    v[12] = .5
-    u = f(v)
-    assert_almost_equal (expected[1][:16], u)
-    # one more time
-    v = fvec(16)
-    u = f(v)
-    assert_almost_equal (expected[1][16:], u)
+    def test_a_weighting(self):
+        expected = array_from_text_file('a_weighting_test_simple.expected')
+        f = digital_filter(7)
+        f.set_a_weighting(44100)
+        v = fvec(32)
+        v[12] = .5
+        u = f(v)
+        assert_almost_equal (expected[1], u)
+
+    def test_a_weighting_parted(self):
+        expected = array_from_text_file('a_weighting_test_simple.expected')
+        f = digital_filter(7)
+        f.set_a_weighting(44100)
+        v = fvec(16)
+        v[12] = .5
+        u = f(v)
+        assert_almost_equal (expected[1][:16], u)
+        # one more time
+        v = fvec(16)
+        u = f(v)
+        assert_almost_equal (expected[1][16:], u)
+
+    def test_set_biquad(self):
+        f = digital_filter(3)
+        f.set_biquad(0., 0., 0, 0., 0.)
+
+    def test_set_biquad_wrong_order(self):
+        f = digital_filter(4)
+        with self.assertRaises(ValueError):
+            f.set_biquad(0., 0., 0, 0., 0.)
+
+class aubio_filter_wrong_params(TestCase):
+
+    def test_negative_order(self):
+        with self.assertRaises(ValueError):
+            digital_filter(-1)
 
 if __name__ == '__main__':
-  from unittest import main
-  main()
-
+    main()