ref: 79287252163131fd0bf41a7c8abb3863cc22677b
dir: /tests/python/cvec.py/
import unittest from aubio.aubiowrapper import * buf_size = 2048 channels = 3 class cvec_test_case(unittest.TestCase): def setUp(self): self.vector = new_cvec(buf_size, channels) def tearDown(self): del_cvec(self.vector) def test_cvec(self): """ create and delete cvec """ pass def test_cvec_read_norm(self): """ check new cvec norm elements are set to 0. """ for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(cvec_read_norm(self.vector,channel,index),0.) def test_cvec_read_phas(self): """ check new cvec phas elements are set to 0. """ for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(cvec_read_phas(self.vector,channel,index),0.) def test_cvec_write_norm(self): """ check new cvec norm elements are set with cvec_write_norm """ for index in range(buf_size/2+1): for channel in range(channels): cvec_write_norm(self.vector,1.,channel,index) for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(cvec_read_norm(self.vector,channel,index),1.) def test_cvec_write_phas(self): """ check new cvec phas elements are set with cvec_write_phas """ for index in range(buf_size/2+1): for channel in range(channels): cvec_write_phas(self.vector,1.,channel,index) for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(cvec_read_phas(self.vector,channel,index),1.) if __name__ == '__main__': unittest.main()