Changeset f91737d

Timestamp:

Oct 3, 2016, 11:46:37 AM (8 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, sampler, yinfft+

Children:

eacc55c

Parents:

80d0083

Message:

python/tests/test_source.py: check the tail of the file is non-zero on non silent test files

File:

• python/tests/test_source.py (modified) (3 diffs)

python/tests/test_source.py

@@ -3 +3 @@
 from nose2 import main
 from nose2.tools import params
-from numpy.testing import TestCase
+from numpy.testing import TestCase, assert_equal
 from aubio import source
 from utils import list_all_sounds
+import numpy as np
 
 import warnings
@@ -52 +53 @@
         total_frames = 0
         while True:
-            _ , read = f()
+            samples , read = f()
             total_frames += read
-            if read < f.hop_size: break
+            if read < f.hop_size:
+                assert_equal(samples[read:], 0)
+                if 'silence' not in f.uri:
+                    self.assertEquals(np.count_nonzero(samples[:read]), read)
+                break
         #result_str = "read {:.2f}s ({:d} frames in {:d} blocks at {:d}Hz) from {:s}"
         #result_params = total_frames / float(f.samplerate), total_frames, total_frames//f.hop_size, f.samplerate, f.uri
@@ -150 +155 @@
         total_frames = 0
         while True:
-            _, read = f.do_multi()
+            samples, read = f.do_multi()
             total_frames += read
-            if read < f.hop_size: break
+            if read < f.hop_size:
+                assert_equal(samples[:,read:], 0)
+                if 'silence' not in f.uri:
+                    self.assertEquals(np.count_nonzero(samples[:,:read]), read)
+                break
         #result_str = "read {:.2f}s ({:d} frames in {:d} channels and {:d} blocks at {:d}Hz) from {:s}"
         #result_params = total_frames / float(f.samplerate), total_frames, f.channels, int(total_frames/f.hop_size), f.samplerate, f.uri