| @@ -0,0 +1,51 @@ |
| 1 | +import unittest |
| 2 | + |
| 3 | +from aubio.aubiowrapper import * |
| 4 | + |
| 5 | +buf_size = 2048 |
| 6 | +channels = 3 |
| 7 | + |
| 8 | +class cvec_test_case(unittest.TestCase): |
| 9 | + |
| 10 | + def setUp(self): |
| 11 | + self.vector = new_cvec(buf_size, channels) |
| 12 | + |
| 13 | + def tearDown(self): |
| 14 | + del_cvec(self.vector) |
| 15 | + |
| 16 | + def test_cvec(self): |
| 17 | + """ create and delete cvec """ |
| 18 | + pass |
| 19 | + |
| 20 | + def test_cvec_read_norm(self): |
| 21 | + """ check new cvec norm elements are set to 0. """ |
| 22 | + for index in range(buf_size/2+1): |
| 23 | + for channel in range(channels): |
| 24 | + self.assertEqual(cvec_read_norm(self.vector,channel,index),0.) |
| 25 | + |
| 26 | + def test_cvec_read_phas(self): |
| 27 | + """ check new cvec phas elements are set to 0. """ |
| 28 | + for index in range(buf_size/2+1): |
| 29 | + for channel in range(channels): |
| 30 | + self.assertEqual(cvec_read_phas(self.vector,channel,index),0.) |
| 31 | + |
| 32 | + def test_cvec_write_norm(self): |
| 33 | + """ check new cvec norm elements are set with cvec_write_norm """ |
| 34 | + for index in range(buf_size/2+1): |
| 35 | + for channel in range(channels): |
| 36 | + cvec_write_norm(self.vector,1.,channel,index) |
| 37 | + for index in range(buf_size/2+1): |
| 38 | + for channel in range(channels): |
| 39 | + self.assertEqual(cvec_read_norm(self.vector,channel,index),1.) |
| 40 | + |
| 41 | + def test_cvec_write_phas(self): |
| 42 | + """ check new cvec phas elements are set with cvec_write_phas """ |
| 43 | + for index in range(buf_size/2+1): |
| 44 | + for channel in range(channels): |
| 45 | + cvec_write_phas(self.vector,1.,channel,index) |
| 46 | + for index in range(buf_size/2+1): |
| 47 | + for channel in range(channels): |
| 48 | + self.assertEqual(cvec_read_phas(self.vector,channel,index),1.) |
| 49 | + |
| 50 | +if __name__ == '__main__': |
| 51 | + unittest.main() |