source: python/tests/test_pitchshift.py @ 0770148

feature/cnnfeature/crepefeature/pitchshiftfeature/timestretchfix/ffmpeg5
Last change on this file since 0770148 was 0770148, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[test] update pitchshift test

  • Property mode set to 100755
File size: 3.6 KB
Line 
1#! /usr/bin/env python
2
3from numpy.testing import TestCase
4from _tools import parametrize
5import numpy as np
6import aubio
7
8class aubio_pitchshift(TestCase):
9
10    def setUp(self):
11        try:
12            self.o = aubio.pitchshift(hop_size = 128)
13        except RuntimeError as e:
14            self.skipTest("creating aubio.pitchshift {}".format(e))
15
16    def test_default_creation(self):
17        self.assertEqual(self.o.get_pitchscale(), 1)
18        self.assertEqual(self.o.get_transpose(), 0)
19
20    def test_on_zeros(self):
21        test_length = self.o.hop_size * 100
22        read = 0
23        # test on zeros
24        vec = aubio.fvec(self.o.hop_size)
25        transpose_range = 24
26        while read < test_length:
27            # transpose the samples
28            out = self.o(vec)
29            self.assertTrue((out == 0).all())
30            # position in the file (between 0. and 1.)
31            percent_read = read / float(test_length)
32            # variable transpose rate (in semitones)
33            transpose = 2 * transpose_range * percent_read - transpose_range
34            # set transpose rate
35            self.o.set_transpose(transpose)
36            read += len(vec)
37
38    def test_on_ones(self):
39        test_length = self.o.hop_size * 100
40        read = 0
41        # test on zeros
42        vec = aubio.fvec(self.o.hop_size) + 1
43        transpose_range = 1.24
44        while read < test_length:
45            # transpose the samples
46            out = self.o(vec)
47            # position in the file (between 0. and 1.)
48            percent_read = read / float(test_length)
49            # variable transpose rate (in semitones)
50            transpose = 2 * transpose_range * percent_read - transpose_range
51            # set transpose rate
52            self.o.set_transpose(transpose)
53            read += len(vec)
54
55    def test_transpose_too_high(self):
56        with self.assertRaises(ValueError):
57            self.o.set_transpose(24.3)
58
59    def test_transpose_too_low(self):
60        with self.assertRaises(ValueError):
61            self.o.set_transpose(-24.3)
62
63class aubio_pitchshift_wrong_params(TestCase):
64
65    def test_wrong_transpose(self):
66        with self.assertRaises(RuntimeError):
67            aubio.pitchshift("default", -123)
68
69class Test_aubio_pitchshift_testruns(object):
70
71    run_args = ['mode', 'pitchscale', 'hop_size', 'samplerate']
72    run_values = [
73            ("default",     1.2,  128,  44100),
74            ("crispness:0", 0.43,  64,   8000),
75            ("crispness:3", 0.53, 256,   8000),
76            ("crispness:3", 1.53, 512,   8000),
77            ("crispness:6", 2.3, 4096, 192000),
78            ]
79
80    @parametrize(run_args, run_values)
81    def test_run_with_params(self, mode, pitchscale, hop_size, samplerate):
82        try:
83            self.o = aubio.pitchshift(mode, pitchscale, hop_size, samplerate)
84        except RuntimeError as e:
85            self.skipTest("failed creating pitchshift ({})".format(e))
86        test_length = self.o.hop_size * 50
87        read = 0
88        # test on random
89        vec = np.random.rand(self.o.hop_size).astype(aubio.float_type)
90        transpose_range = self.o.get_transpose()
91        while read < test_length:
92            # transpose the samples
93            out = self.o(vec)
94            # position in the file (between 0. and 1.)
95            percent_read = read / float(test_length)
96            # variable transpose rate (in semitones)
97            transpose =  transpose_range - 2 * transpose_range * percent_read
98            # set transpose rate
99            self.o.set_transpose(transpose)
100            read += len(vec)
101
102if __name__ == '__main__':
103    from _tools import run_module_suite
104    run_module_suite()
Note: See TracBrowser for help on using the repository browser.