add test for fft (which appears to reveal a bug in fftw on powerpc\!)

diff --git a/tests/python/fft.py b/tests/python/fft.py
new file mode 100644 (file)
index 0000000..540a6c2
--- /dev/null
+++ b/tests/python/fft.py
@@ -0,0 +1,105 @@
+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()