Changeset a0e0f56

Timestamp:

Sep 21, 2016, 1:32:57 AM (8 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn, feature/crepe, feature/pitchshift, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch

Children:

04fc360

Parents:

b3f79ca

Message:

python/tests/test_pitchshift.py: run in a few modes

File:

• python/tests/test_pitchshift.py (modified) (5 diffs)

python/tests/test_pitchshift.py

@@ -2 +2 @@
 
 from numpy.testing import TestCase
+from nose2.tools import params
+import numpy as np
 import aubio
 
@@ -8 +10 @@
     def setUp(self):
         try:
-            self.o = aubio.pitchshift()
+            self.o = aubio.pitchshift(hop_size = 128)
         except RuntimeError as e:
             self.skipTest("creating aubio.pitchshift failed (recompile with rubberband?)")
@@ -17 +19 @@
 
     def test_on_zeros(self):
-        test_length = 20000
+        test_length = self.o.hop_size * 100
         read = 0
         # test on zeros
-        vec = aubio.fvec(512)
+        vec = aubio.fvec(self.o.hop_size)
         transpose_range = 24
         while read < test_length:
@@ -26 +28 @@
             out = self.o(vec)
             self.assertTrue((out == 0).all())
+            # position in the file (between 0. and 1.)
+            percent_read = read / float(test_length)
+            # variable transpose rate (in semitones)
+            transpose = 2 * transpose_range * percent_read - transpose_range
+            # set transpose rate
+            self.o.set_transpose(transpose)
+            read += len(vec)
+
+    def test_on_ones(self):
+        test_length = self.o.hop_size * 100
+        read = 0
+        # test on zeros
+        vec = aubio.fvec(self.o.hop_size) + 1
+        transpose_range = 1.24
+        while read < test_length:
+            # transpose the samples
+            out = self.o(vec)
             # position in the file (between 0. and 1.)
             percent_read = read / float(test_length)
@@ -42 +61 @@
             self.o.set_transpose(-24.3)
 
+
+class aubio_pitchshift_testruns(TestCase):
+
+    @params(
+            ("default",     1.2,  128,  44100),
+            ("crispness:0", 0.43,  64,   8000),
+            ("crispness:3", 0.53, 256,   8000),
+            ("crispness:3", 1.53, 512,   8000),
+            ("crispness:6", 2.3, 4096, 192000),
+            )
+    def test_run_with_params(self, mode, pitchscale, hop_size, samplerate):
+        self.o = aubio.pitchshift(mode, pitchscale, hop_size, samplerate)
+        test_length = self.o.hop_size * 50
+        read = 0
+        # test on random
+        vec = np.random.rand(self.o.hop_size).astype(aubio.float_type)
+        transpose_range = self.o.get_transpose()
+        while read < test_length:
+            # transpose the samples
+            out = self.o(vec)
+            # position in the file (between 0. and 1.)
+            percent_read = read / float(test_length)
+            # variable transpose rate (in semitones)
+            transpose =  transpose_range - 2 * transpose_range * percent_read
+            # set transpose rate
+            self.o.set_transpose(transpose)
+            read += len(vec)
+
 if __name__ == '__main__':
     from nose2 import main