source: python/tests/test_pitch.py @ c3de7ca

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5
Last change on this file since c3de7ca was c3de7ca, checked in by Paul Brossier <piem@piem.org>, 7 years ago

python/tests/test_pitch.py: add yinfast

  • Property mode set to 100755
File size: 4.5 KB
Line 
1#! /usr/bin/env python
2
3from unittest import TestCase, main
4from numpy.testing import assert_equal
5from numpy import sin, arange, mean, median, isnan, pi
6from aubio import fvec, pitch, freqtomidi, float_type
7
8class aubio_pitch_Good_Values(TestCase):
9
10    def skip_test_new_default(self):
11        " creating a pitch object without parameters "
12        p = pitch()
13        assert_equal ( [p.method, p.buf_size, p.hop_size, p.samplerate],
14            ['default', 1024, 512, 44100])
15
16    def test_run_on_silence(self):
17        " creating a pitch object with parameters "
18        p = pitch('default', 2048, 512, 32000)
19        assert_equal ( [p.method, p.buf_size, p.hop_size, p.samplerate],
20            ['default', 2048, 512, 32000])
21
22    def test_run_on_zeros(self):
23        " running on silence gives 0 "
24        p = pitch('default', 2048, 512, 32000)
25        f = fvec (512)
26        for _ in range(10): assert_equal (p(f), 0.)
27
28    def test_run_on_ones(self):
29        " running on ones gives 0 "
30        p = pitch('default', 2048, 512, 32000)
31        f = fvec (512)
32        f[:] = 1
33        for _ in range(10): assert_equal (p(f), 0.)
34
35class aubio_pitch_Sinusoid(TestCase):
36
37    def run_pitch_on_sinusoid(self, method, buf_size, hop_size, samplerate, freq):
38        # create pitch object
39        p = pitch(method, buf_size, hop_size, samplerate)
40        # duration in seconds
41        seconds = .3
42        # duration in samples
43        duration =  seconds * samplerate
44        # increase to the next multiple of hop_size
45        duration = duration - duration % hop_size + hop_size;
46        # build sinusoid
47        sinvec = self.build_sinusoid(duration, freq, samplerate)
48
49        self.run_pitch(p, sinvec, freq)
50
51    def build_sinusoid(self, length, freq, samplerate):
52        return sin( 2. * pi * arange(length).astype(float_type) * freq / samplerate)
53
54    def run_pitch(self, p, input_vec, freq):
55        pitches, errors = [], []
56        input_blocks = input_vec.reshape((-1, p.hop_size))
57        for new_block in input_blocks:
58            pitch = p(new_block)[0]
59            pitches.append(pitch)
60            errors.append(1. - freqtomidi(pitch) / freqtomidi(freq))
61        assert_equal ( len(input_blocks), len(pitches) )
62        assert_equal ( isnan(pitches), False )
63        # cut the first candidates
64        #cut = ( p.buf_size - p.hop_size ) / p.hop_size
65        pitches = pitches[2:]
66        errors = errors[2:]
67        # check that the mean of all relative errors is less than 10%
68        #assert abs (mean(errors) ) < 0.1, pitches
69        assert abs (median(errors) ) < 0.06, "median of relative errors is bigger than 0.06 (%f)\n found %s\n errors %s" % (mean(errors), pitches, errors)
70        #print 'len(pitches), cut:', len(pitches), cut
71        #print 'median errors: ', median(errors), 'median pitches: ', median(pitches)
72
73pitch_algorithms = [ "default", "yinfft", "yin", "yinfast", "schmitt", "mcomb", "fcomb" , "specacf" ]
74pitch_algorithms = [ "default", "yinfft", "yin", "yinfast", "schmitt", "mcomb", "fcomb" ]
75
76#freqs = [ 27.5, 55., 110., 220., 440., 880., 1760., 3520. ]
77freqs = [             110., 220., 440., 880., 1760., 3520. ]
78signal_modes = []
79for freq in freqs:
80    signal_modes += [
81        ( 4096,  2048, 44100, freq ),
82        ( 2048,  512, 44100, freq ),
83        ( 2048, 1024, 44100, freq ),
84        ( 2048, 1024, 32000, freq ),
85        ]
86
87freqs = [ ] #55., 110., 220., 440., 880., 1760., 3520. ]
88for freq in freqs:
89    signal_modes += [
90        ( 2048, 1024, 22050, freq ),
91        ( 1024,  256, 16000, freq ),
92        ( 1024,  256, 8000,  freq ),
93        ( 1024, 512+256, 8000, freq ),
94        ]
95
96"""
97signal_modes = [
98        ( 4096,  512, 44100, 2.*882. ),
99        ( 4096,  512, 44100, 882. ),
100        ( 4096,  512, 44100, 440. ),
101        ( 2048,  512, 44100, 440. ),
102        ( 2048, 1024, 44100, 440. ),
103        ( 2048, 1024, 44100, 440. ),
104        ( 2048, 1024, 32000, 440. ),
105        ( 2048, 1024, 22050, 440. ),
106        ( 1024,  256, 16000, 440. ),
107        ( 1024,  256, 8000,  440. ),
108        ( 1024, 512+256, 8000, 440. ),
109        ]
110"""
111
112def create_test (algo, mode):
113    def do_test_pitch(self):
114        self.run_pitch_on_sinusoid(algo, mode[0], mode[1], mode[2], mode[3])
115    return do_test_pitch
116
117for algo in pitch_algorithms:
118    for mode in signal_modes:
119        test_method = create_test (algo, mode)
120        test_method.__name__ = 'test_pitch_%s_%d_%d_%dHz_sin_%.0f' % ( algo,
121                mode[0], mode[1], mode[2], mode[3] )
122        setattr (aubio_pitch_Sinusoid, test_method.__name__, test_method)
123
124if __name__ == '__main__':
125    main()
Note: See TracBrowser for help on using the repository browser.