python/tests/test_cvec.py

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