ref: a2f3555489f509d3c223f44a356da960f9974a5f
dir: /tests/python/phasevoc.py/
import unittest from aubio.aubiowrapper import * buf_size = 1024 hop_size = 256 channels = 2 precision = 6 class phasevoc_unit(unittest.TestCase): def setUp(self): self.o = new_aubio_pvoc(buf_size, hop_size, channels) def tearDown(self): del_aubio_pvoc(self.o) def test_create(self): """ create and delete phasevoc object """ pass def test_zeroes(self): """ run phasevoc object on zeroes """ input = new_fvec(hop_size, channels) fftgrain = new_cvec(buf_size, channels) output = new_fvec(hop_size, channels) for index in range(hop_size): for channel in range(channels): self.assertEqual(0., fvec_read_sample(input, channel, index)) aubio_pvoc_do (self.o, input, fftgrain) aubio_pvoc_rdo(self.o, fftgrain, output) for index in range(hop_size): for channel in range(channels): self.assertEqual(0., fvec_read_sample(output, channel, index)) del input del fftgrain def test_ones(self): """ run phasevoc object on ones """ input = new_fvec(hop_size, channels) fftgrain = new_cvec(buf_size, channels) output = new_fvec(hop_size, channels) for index in range(hop_size): for channel in range(channels): fvec_write_sample(input, 1., channel, index) self.assertEqual(1., fvec_read_sample(input, channel, index)) # make sure the first buf_size-hop_size samples are zeroes for i in range(buf_size/hop_size - 1): aubio_pvoc_do (self.o, input, fftgrain) aubio_pvoc_rdo(self.o, fftgrain, output) for index in range(hop_size): for channel in range(channels): self.assertAlmostEqual(0., fvec_read_sample(output, channel, index), precision) # make sure the first non zero input is correctly resynthesised aubio_pvoc_do (self.o, input, fftgrain) aubio_pvoc_rdo(self.o, fftgrain, output) for index in range(hop_size): for channel in range(channels): self.assertAlmostEqual(1., fvec_read_sample(output, channel, index), precision) del input del fftgrain