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() |
---|