Changeset c09efca

Timestamp:

May 13, 2016, 5:58:02 PM (9 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:

1167631

Parents:

1164fd85

Message:

python/tests/test_cvec.py: add more tests

File:

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

python/tests/test_cvec.py

@@ -5 +5 @@
 from aubio import cvec, fvec, float_type
 import numpy as np
+
+wrong_type = 'float32' if float_type == 'float64' else 'float64'
 
 class aubio_cvec_test_case(TestCase):
@@ -50 +52 @@
         """ check dest cvec is still reachable after source was deleted """
         spec = cvec(1024)
-        a = np.random.rand(1024/2+1).astype(float_type)
-        b = np.random.rand(1024/2+1).astype(float_type)
+        a = np.random.rand(1024//2+1).astype(float_type)
+        b = np.random.rand(1024//2+1).astype(float_type)
         spec.norm = a
         spec.phas = b
@@ -103 +105 @@
             a.phas = b
 
+    def test_cvec_repr(self):
+        win_s = 512
+        c = cvec(win_s)
+        expected_repr = "aubio cvec of {:d} elements".format(win_s//2+1)
+        self.assertEqual(repr(c), expected_repr)
+
+class aubio_cvec_wrong_norm_input(TestCase):
+
+    def test_wrong_length(self):
+        with self.assertRaises(ValueError):
+            cvec(-1)
+
+    def test_set_norm_with_scalar(self):
+        a = cvec(512)
+        with self.assertRaises(ValueError):
+            a.norm = 1
+
+    def test_set_norm_with_scalar_array(self):
+        a = cvec(512)
+        with self.assertRaises(ValueError):
+            a.norm = np.ndarray(1, dtype = 'int')
+
+    def test_set_norm_with_int_array(self):
+        a = cvec(512)
+        with self.assertRaises(ValueError):
+            a.norm = np.zeros(512//2+1, dtype = 'int')
+
+    def test_set_norm_with_wrong_float_array(self):
+        a = cvec(512)
+        with self.assertRaises(ValueError):
+            a.norm = np.zeros(512//2+1, dtype = wrong_type)
+
+    def test_set_norm_with_wrong_2d_array(self):
+        a = cvec(512)
+        with self.assertRaises(ValueError):
+            a.norm = np.zeros((512//2+1, 2), dtype = float_type)
+
 if __name__ == '__main__':
     from nose2 import main