python/tests/test_cvec.py

   1 #! /usr/bin/env python
   2
   3 from numpy.testing import TestCase
   4 from numpy.testing import assert_equal, assert_almost_equal
   5 from aubio import cvec, fvec, float_type
   6 import numpy as np
   7
   8 wrong_type = 'float32' if float_type == 'float64' else 'float64'
   9
  10 class aubio_cvec_test_case(TestCase):
  11
  12     def test_vector_created_with_zeroes(self):
  13         a = cvec(10)
  14         assert_equal(a.norm.shape[0], 10 / 2 + 1)
  15         assert_equal(a.phas.shape[0], 10 / 2 + 1)
  16         a.norm[0]
  17         assert_equal(a.norm, 0.)
  18         assert_equal(a.phas, 0.)
  19
  20     def test_vector_assign_element(self):
  21         a = cvec()
  22         a.norm[0] = 1
  23         assert_equal(a.norm[0], 1)
  24         a.phas[0] = 1
  25         assert_equal(a.phas[0], 1)
  26
  27     def test_vector_assign_element_end(self):
  28         a = cvec()
  29         a.norm[-1] = 1
  30         assert_equal(a.norm[-1], 1)
  31         assert_equal(a.norm[len(a.norm)-1], 1)
  32         a.phas[-1] = 1
  33         assert_equal(a.phas[-1], 1)
  34         assert_equal(a.phas[len(a.phas)-1], 1)
  35
  36     def test_assign_cvec_norm_slice(self):
  37         spec = cvec(1024)
  38         spec.norm[40:100] = 100
  39         assert_equal(spec.norm[0:40], 0)
  40         assert_equal(spec.norm[40:100], 100)
  41         assert_equal(spec.norm[100:-1], 0)
  42         assert_equal(spec.phas, 0)
  43
  44     def test_assign_cvec_phas_slice(self):
  45         spec = cvec(1024)
  46         spec.phas[39:-1] = -np.pi
  47         assert_equal(spec.phas[0:39], 0)
  48         assert_equal(spec.phas[39:-1], -np.pi)
  49         assert_equal(spec.norm, 0)
  50
  51     def test_assign_cvec_with_other_cvec(self):
  52         """ check dest cvec is still reachable after source was deleted """
  53         spec = cvec(1024)
  54         a = np.random.rand(1024//2+1).astype(float_type)
  55         b = np.random.rand(1024//2+1).astype(float_type)
  56         spec.norm = a
  57         spec.phas = b
  58         new_spec = spec
  59         del spec
  60         assert_equal(a, new_spec.norm)
  61         assert_equal(b, new_spec.phas)
  62         assert_equal(id(a), id(new_spec.norm))
  63         assert_equal(id(b), id(new_spec.phas))
  64
  65     def test_pass_to_numpy(self):
  66         spec = cvec(1024)
  67         norm = spec.norm
  68         phas = spec.phas
  69         del spec
  70         new_spec = cvec(1024)
  71         new_spec.norm = norm
  72         new_spec.phas = phas
  73         assert_equal(norm, new_spec.norm)
  74         assert_equal(phas, new_spec.phas)
  75         assert_equal(id(norm), id(new_spec.norm))
  76         assert_equal(id(phas), id(new_spec.phas))
  77         del norm
  78         del phas
  79         assert_equal(new_spec.norm, 0.)
  80         assert_equal(new_spec.phas, 0.)
  81         del new_spec
  82
  83     def test_assign_norm_too_large(self):
  84         a = cvec(512)
  85         b = fvec(512//2+1 + 4)
  86         with self.assertRaises(ValueError):
  87             a.norm = b
  88
  89     def test_assign_norm_too_small(self):
  90         a = cvec(512)
  91         b = fvec(512//2+1 - 4)
  92         with self.assertRaises(ValueError):
  93             a.norm = b
  94
  95     def test_assign_phas_too_large(self):
  96         a = cvec(512)
  97         b = fvec(512//2+1 + 4)
  98         with self.assertRaises(ValueError):
  99             a.phas = b
 100
 101     def test_assign_phas_too_small(self):
 102         a = cvec(512)
 103         b = fvec(512//2+1 - 4)
 104         with self.assertRaises(ValueError):
 105             a.phas = b
 106
 107     def test_cvec_repr(self):
 108         win_s = 512
 109         c = cvec(win_s)
 110         expected_repr = "aubio cvec of {:d} elements".format(win_s//2+1)
 111         self.assertEqual(repr(c), expected_repr)
 112
 113 class aubio_cvec_wrong_norm_input(TestCase):
 114
 115     def test_wrong_length(self):
 116         with self.assertRaises(ValueError):
 117             cvec(-1)
 118
 119     def test_set_norm_with_scalar(self):
 120         a = cvec(512)
 121         with self.assertRaises(ValueError):
 122             a.norm = 1
 123
 124     def test_set_norm_with_scalar_array(self):
 125         a = cvec(512)
 126         with self.assertRaises(ValueError):
 127             a.norm = np.ndarray(1, dtype = 'int')
 128
 129     def test_set_norm_with_int_array(self):
 130         a = cvec(512)
 131         with self.assertRaises(ValueError):
 132             a.norm = np.zeros(512//2+1, dtype = 'int')
 133
 134     def test_set_norm_with_wrong_float_array(self):
 135         a = cvec(512)
 136         with self.assertRaises(ValueError):
 137             a.norm = np.zeros(512//2+1, dtype = wrong_type)
 138
 139     def test_set_norm_with_wrong_2d_array(self):
 140         a = cvec(512)
 141         with self.assertRaises(ValueError):
 142             a.norm = np.zeros((512//2+1, 2), dtype = float_type)
 143
 144 if __name__ == '__main__':
 145     from nose2 import main
 146     main()