ref: 6e157df38d1a9acb48810660b7b8f1a9e7d87a7e
dir: /python/tests/test_pitchshift.py/
#! /usr/bin/env python
from numpy.testing import TestCase
from nose2.tools import params
import numpy as np
import aubio
class aubio_pitchshift(TestCase):
def setUp(self):
try:
self.o = aubio.pitchshift(hop_size = 128)
except RuntimeError as e:
self.skipTest("creating aubio.pitchshift {}".format(e))
def test_default_creation(self):
self.assertEqual(self.o.get_pitchscale(), 1)
self.assertEqual(self.o.get_transpose(), 0)
def test_on_zeros(self):
test_length = self.o.hop_size * 100
read = 0
# test on zeros
vec = aubio.fvec(self.o.hop_size)
transpose_range = 24
while read < test_length:
# transpose the samples
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)
# 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_transpose_too_high(self):
with self.assertRaises(ValueError):
self.o.set_transpose(24.3)
def test_transpose_too_low(self):
with self.assertRaises(ValueError):
self.o.set_transpose(-24.3)
class aubio_pitchshift_wrong_params(TestCase):
def test_wrong_transpose(self):
with self.assertRaises(RuntimeError):
aubio.pitchshift("default", -123)
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):
try:
self.o = aubio.pitchshift(mode, pitchscale, hop_size, samplerate)
except RuntimeError as e:
self.skipTest("failed creating pitchshift ({})".format(e))
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
main()