Changeset e1bfde5 for python/tests/test_source.py

Timestamp:

Mar 6, 2013, 9:25:23 PM (11 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:

037319a

Parents:

143cc43

Message:

tests/test_{source,sink}.py: add sink, improve source, use sounds in python/tests/sounds

File:

• python/tests/test_source.py (modified) (1 diff)

python/tests/test_source.py

@@ -4 +4 @@
 from aubio import fvec, source
 from numpy import array
+from utils import list_all_sounds
 
-path = "/Users/piem/archives/sounds/loops/drum_Chocolate_Milk_-_Ation_Speaks_Louder_Than_Words.wav"
+list_of_sounds = list_all_sounds('sounds')
+path = None
 
-class aubio_filter_test_case(TestCase):
+class aubio_source_test_case(TestCase):
 
-  def test_members(self):
-    f = source(path)
-    print dir(f)
+    def setUp(self):
+        if not len(list_of_sounds): self.skipTest('add some sound files in \'python/tests/sounds\'')
 
-  def test_read(self):
-    f = source(path)
-    total_frames = 0
-    while True:
-      vec, read = f()
-      total_frames += read
-      if read < f.hop_size: break
-    print "read", total_frames / float(f.samplerate), " seconds from", path
+    def read_from_sink(self, f):
+        total_frames = 0
+        while True:
+            vec, read = f()
+            total_frames += read
+            if read < f.hop_size: break
+        print "read", "%.2fs" % (total_frames / float(f.samplerate) ),
+        print "(", total_frames, "frames", "in",
+        print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")",
+        print "from", f.uri
+
+    def test_samplerate_hopsize(self):
+        for p in list_of_sounds:
+            for samplerate, hop_size in zip([0, 44100, 8000, 32000], [ 512, 512, 64, 256]):
+                f = source(p, samplerate, hop_size)
+                assert f.samplerate != 0
+                self.read_from_sink(f)
+
+    def test_samplerate_none(self):
+        for p in list_of_sounds:
+            f = source(p)
+            assert f.samplerate != 0
+            self.read_from_sink(f)
+
+    def test_samplerate_0(self):
+        for p in list_of_sounds:
+            f = source(p, 0)
+            assert f.samplerate != 0
+            self.read_from_sink(f)
+
+    def test_wrong_samplerate(self):
+        for p in list_of_sounds:
+            try:
+                f = source(p, -1)
+            except Exception, e:
+                print e
+            else:
+                self.fail('does not fail with wrong samplerate')
+
+    def test_wrong_hop_size(self):
+        for p in list_of_sounds:
+            f = source(p, 0, -1)
+            print f.hop_size
+
+    def test_zero_hop_size(self):
+        for p in list_of_sounds:
+            f = source(p, 0, 0)
+            assert f.samplerate != 0
+            assert f.hop_size != 0
+            self.read_from_sink(f)
 
 if __name__ == '__main__':
-  from unittest import main
-  main()
-
+    from unittest import main
+    main()