Changeset 1594d49 for tests

Timestamp:

Oct 8, 2009, 2:00:52 AM (15 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

8fe9c10

Parents:

db62622

Message:

tests/python: updated filterbank test

Location:

tests/python/src/spectral

Files:

• filterbank.py (modified) (7 diffs)
• filterbank_mfcc.txt (deleted)
• filterbank_mfcc_16000_512.txt (added)

tests/python/src/spectral/filterbank.py

@@ -1 @@
-
-import unittest
-from aubio.aubiowrapper import * 
+from template import aubio_unit_template
+from localaubio import *
 
 win_size = 2048 
@@ -7 +6 @@
 n_filters = 40
 samplerate = 44100
-zerodb = -96.015602111816406
 
 class filterbank_test_case(unittest.TestCase):
@@ -22 +20 @@
 
   def testzeroes(self):
-      """ check the output of the filterbank is -96 when input spectrum is 0 """
+      """ check the output of the filterbank is 0 when input spectrum is 0 """
       aubio_filterbank_do(self.filterbank,self.input_spectrum,
         self.output_banks) 
       for channel in range(channels):
         for index in range(n_filters): 
-          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), zerodb)
+          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), 0)
 
   def testphase(self):
-      """ check the output of the filterbank is -96 when input phase is pi """
+      """ check the output of the filterbank is 0 when input phase is pi """
       from math import pi
       for channel in range(channels):
@@ -39 +37 @@
       for channel in range(channels):
         for index in range(n_filters): 
-          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), zerodb)
+          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), 0)
 
   def testones(self):
-      """ check the output of the filterbank is -96 when input norm is 1
+      """ check the output of the filterbank is 0 when input norm is 1
           (the filterbank is currently set to 0).
       """
@@ -52 +50 @@
       for channel in range(channels):
         for index in range(n_filters): 
-          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), zerodb)
+          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), 0)
 
   def testmfcc_zeroes(self):
-      """ check the mfcc filterbank output is -96 when input is 0 """
-      self.filterbank = new_aubio_filterbank_mfcc(n_filters, win_size, samplerate, 0., samplerate) 
-      aubio_filterbank_do(self.filterbank,self.input_spectrum,
+      """ check the mfcc filterbank output is 0 when input is 0 """
+      self.filterbank = new_aubio_filterbank(n_filters, win_size) 
+      aubio_filterbank_do(self.filterbank, self.input_spectrum,
         self.output_banks) 
       for channel in range(channels):
         for index in range(n_filters): 
-          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), zerodb)
+          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), 0)
 
   def testmfcc_phasepi(self):
-      """ check the mfcc filterbank output is -96 when input phase is pi """
-      self.filterbank = new_aubio_filterbank_mfcc(n_filters, win_size, samplerate, 0., samplerate) 
+      """ check the mfcc filterbank output is 0 when input phase is pi """
+      self.filterbank = new_aubio_filterbank(n_filters, win_size) 
+      aubio_filterbank_set_mel_coeffs_slaney(self.filterbank, samplerate)
       from math import pi
       for channel in range(channels):
@@ -74 +73 @@
       for channel in range(channels):
         for index in range(n_filters): 
-          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), zerodb)
+          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), 0)
 
   def testmfcc_ones(self):
-      """ check setting the input spectrum to 1 gives something between -3. and -4. """ 
-      self.filterbank = new_aubio_filterbank_mfcc(n_filters, win_size, samplerate, 0., samplerate) 
+      """ check setting the input spectrum to 1 gives something between 0. and 1. """ 
+      self.filterbank = new_aubio_filterbank(n_filters, win_size) 
+      aubio_filterbank_set_mel_coeffs_slaney(self.filterbank, samplerate)
       for channel in range(channels):
         for index in range(win_size/2+1): 
@@ -87 +87 @@
         for index in range(n_filters): 
           val = fvec_read_sample(self.output_banks,channel,index)
-          self.failIf(val > -2.5 , val )
-          self.failIf(val < -4. , val )
+          self.failIf(val < 0. , val )
+          self.failIf(val > 1. , val )
 
   def testmfcc_channels(self):
       """ check the values of each filters in the mfcc filterbank """
       import os.path
-      self.filterbank = new_aubio_filterbank_mfcc(n_filters, win_size, samplerate, 
-        0., samplerate) 
-      filterbank_mfcc = [ [float(f) for f in line.strip().split()]
-        for line in open(os.path.join('src','spectral','filterbank_mfcc.txt')).readlines()]
+      win_size = 512
+      self.filterbank = new_aubio_filterbank(n_filters, win_size) 
+      aubio_filterbank_set_mel_coeffs_slaney(self.filterbank, 16000)
+      filterbank_mfcc = array_from_text_file ( 
+          os.path.join('src','spectral','filterbank_mfcc_16000_512.txt') )
+      vec = aubio_filterbank_get_coeffs(self.filterbank)
       for channel in range(n_filters):
-        vec = aubio_filterbank_getchannel(self.filterbank,channel)
-        for index in range(win_size): 
-          self.assertAlmostEqual(fvec_read_sample(vec,0,index),
+        for index in range(win_size/2+1): 
+          self.assertAlmostEqual(fvec_read_sample(vec,channel,index),
             filterbank_mfcc[channel][index])
-      aubio_filterbank_do(self.filterbank,self.input_spectrum,
+      aubio_filterbank_do(self.filterbank, self.input_spectrum,
         self.output_banks) 
       for channel in range(channels):
         for index in range(n_filters): 
-          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), zerodb)
+          self.assertEqual(fvec_read_sample(self.output_banks,channel,index), 0)
 
 if __name__ == '__main__':