ref: 398c1c5c18d85587e67ad6d4bee3f25ce37bf437
dir: /tests/python/fft.py/
import unittest import math from aubio.aubiowrapper import * buf_size = 2048 channels = 1 class aubio_mfft_test_case(unittest.TestCase): def setUp(self): self.o = new_aubio_mfft(buf_size, channels) def tearDown(self): del_aubio_mfft(self.o) def test_aubio_mfft(self): """ create and delete mfft """ pass def test_aubio_mfft_do_zeroes(self): """ test aubio_mfft_do on zeroes """ input = new_fvec(buf_size, channels) fftgrain = new_cvec(buf_size, channels) for index in range(buf_size): for channel in range(channels): self.assertEqual(0., fvec_read_sample(input, channel, index)) aubio_mfft_do(self.o, input, fftgrain) for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(0., cvec_read_phas(fftgrain, channel, index)) del fftgrain del input def test_aubio_mfft_rdo_zeroes(self): """ test aubio_mfft_rdo on zeroes """ fftgrain = new_cvec(buf_size, channels) output = new_fvec(buf_size, channels) aubio_mfft_rdo(self.o, fftgrain, output) # check output for index in range(buf_size): for channel in range(channels): self.assertEqual(0., fvec_read_sample(output, channel, index)) del fftgrain del output def test_aubio_mfft_do_impulse(self): """ test aubio_mfft_do on impulse one channel """ input = new_fvec(buf_size, channels) fftgrain = new_cvec(buf_size, channels) # write impulse in channel 0, sample 0. fvec_write_sample(input, 1., 0, 0) aubio_mfft_do(self.o, input, fftgrain) # check norm for index in range(buf_size/2+1): self.assertEqual(1., cvec_read_norm(fftgrain, 0, index), index) for index in range(buf_size/2+1): for channel in range(1, channels): self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) # check phas for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(0., cvec_read_phas(fftgrain, channel, index)) del fftgrain del input def test_aubio_mfft_do_impulse_multichannel(self): " test aubio_mfft_do on impulse two channels " input = new_fvec(buf_size, channels) fftgrain = new_cvec(buf_size, channels) # put an impulse in first an last channel, at first and last index fvec_write_sample(input, 1., 0, 0) fvec_write_sample(input, 1., channels-1, 0) aubio_mfft_do(self.o, input, fftgrain) # check the norm for index in range(buf_size/2+1): self.assertEqual(1., cvec_read_norm(fftgrain, 0, index)) for index in range(buf_size/2+1): for channel in range(1, channels-1): self.assertEqual(0., cvec_read_norm(fftgrain, channel, index)) for index in range(buf_size/2+1): self.assertEqual(1., cvec_read_norm(fftgrain, channels-1, index)) # check the phase for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(0., cvec_read_phas(fftgrain, channel, index)) del fftgrain del input def test_aubio_mfft_rdo_impulse(self): """ test aubio_mfft_rdo on impulse """ fftgrain = new_cvec(buf_size, channels) cvec_write_norm(fftgrain, 1., 0, 0) output = new_fvec(buf_size, channels) aubio_mfft_rdo(self.o, fftgrain, output) for index in range(buf_size/2+1): for channel in range(channels): self.assertEqual(fvec_read_sample(output, channel, index),1./buf_size) del fftgrain del output if __name__ == '__main__': unittest.main()