Merge branch 'master' of https://github.com/DrMarc/slab

DrMarc · Apr 21, 2022 · 1db2a79 · 1db2a79
2 parents 6ecfc0c + 54b1800
commit 1db2a79
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/slab/filter.py b/slab/filter.py
@@ -457,7 +457,7 @@ def equalizing_filterbank(reference, sound, length=1000, bandwidth=1/8, low_cuto
         filts = numpy.zeros((length, sound.n_channels))
         for chan in range(sound.n_channels):
             gain = [1] + list(amp_diffs[:, chan])
-            filt = Filter.band(frequency=list(center_freqs), gain=gain, samplerate=reference.samplerate, fir=True)
+            filt = Filter.band(frequency=list(center_freqs), gain=gain, samplerate=reference.samplerate, fir=True, length=length)
             filts[:, chan] = filt.data.flatten()
         return Filter(data=filts, samplerate=reference.samplerate, fir=True)
 

diff --git a/tests/test_filter.py b/tests/test_filter.py
@@ -101,6 +101,9 @@ def test_center_freqs():
 
 def test_equalization():
     for i in range(10):
+        length = numpy.random.randint(1000, 5000)
+        low_cutoff = numpy.random.randint(20, 2000)
+        high_cutoff = numpy.random.randint(10000, 20000)
         sound = slab.Sound.pinknoise(samplerate=44100)
         filt = slab.Filter.band(frequency=[100., 800., 2000., 4300., 8000., 14500., 18000.],
                                 gain=[0., 1., 0., 1., 0., 1., 0.], samplerate=sound.samplerate)