ref: e6b2a0c83c5a053ead70179e75c0c41aa678b4e7
dir: /tests/python/src/temporal/a_weighting.py/
from localaubio import * samplerate = 44100 buf_size = 1024 channels = 1 class a_weighting_unit (aubio_unit_template): def setUp(self): self.o = new_aubio_filter_a_weighting (samplerate, channels) def tearDown(self): del_aubio_filter (self.o) def test_creation(self): pass def test_filter_zeroes(self): """ check filter run on a vector full of zeroes returns zeros """ vec = new_fvec(buf_size, channels) aubio_filter_do (self.o, vec) for index in range(buf_size): for channel in range(channels): self.assertEqual(0., fvec_read_sample(vec,channel,index)) del_fvec(vec) def test_filter_ones(self): vec = new_fvec(buf_size, channels) for index in range(buf_size): for channel in range(channels): fvec_write_sample(vec, 1., channel, index) aubio_filter_do (self.o, vec) for index in range(buf_size): for channel in range(channels): self.assertNotEqual(0., fvec_read_sample(vec,channel,index)) del_fvec(vec) def test_filter_denormal(self): vec = new_fvec(buf_size, channels) for index in range(buf_size): for channel in range(channels): fvec_write_sample(vec, 2.e-42, channel, index) aubio_filter_do (self.o, vec) for index in range(buf_size): for channel in range(channels): self.assertEqual(0., fvec_read_sample(vec,channel,index)) del_fvec(vec) def test_simple(self): buf_size = 32 input = new_fvec (buf_size, 1) output = new_fvec (buf_size, 1) expected = array_from_text_file('src/temporal/a_weighting_test_simple.expected') fvec_write_sample (input, 0.5, 0, 12) for i in range(buf_size): for c in range(channels): self.assertEqual(expected[0][i], fvec_read_sample(input, c, i)) f = new_aubio_filter_a_weighting (samplerate, channels) aubio_filter_do_outplace (f, input, output) del_aubio_filter (f) for i in range(buf_size): for c in range(channels): self.assertCloseEnough(expected[1][i], fvec_read_sample(output, c, i)) if __name__ == '__main__': import unittest unittest.main()