source: python/tests/test_source.py @ e1bfde5

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

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

  • Property mode set to 100755
File size: 2.1 KB
RevLine 
[75e715f]1#! /usr/bin/env python
[6192ce7]2
3from numpy.testing import TestCase, assert_equal, assert_almost_equal
4from aubio import fvec, source
5from numpy import array
[e1bfde5]6from utils import list_all_sounds
[6192ce7]7
[e1bfde5]8list_of_sounds = list_all_sounds('sounds')
9path = None
[6192ce7]10
[e1bfde5]11class aubio_source_test_case(TestCase):
[6192ce7]12
[e1bfde5]13    def setUp(self):
14        if not len(list_of_sounds): self.skipTest('add some sound files in \'python/tests/sounds\'')
[6192ce7]15
[e1bfde5]16    def read_from_sink(self, f):
17        total_frames = 0
18        while True:
19            vec, read = f()
20            total_frames += read
21            if read < f.hop_size: break
22        print "read", "%.2fs" % (total_frames / float(f.samplerate) ),
23        print "(", total_frames, "frames", "in",
24        print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")",
25        print "from", f.uri
[6192ce7]26
[e1bfde5]27    def test_samplerate_hopsize(self):
28        for p in list_of_sounds:
29            for samplerate, hop_size in zip([0, 44100, 8000, 32000], [ 512, 512, 64, 256]):
30                f = source(p, samplerate, hop_size)
31                assert f.samplerate != 0
32                self.read_from_sink(f)
33
34    def test_samplerate_none(self):
35        for p in list_of_sounds:
36            f = source(p)
37            assert f.samplerate != 0
38            self.read_from_sink(f)
39
40    def test_samplerate_0(self):
41        for p in list_of_sounds:
42            f = source(p, 0)
43            assert f.samplerate != 0
44            self.read_from_sink(f)
45
46    def test_wrong_samplerate(self):
47        for p in list_of_sounds:
48            try:
49                f = source(p, -1)
50            except Exception, e:
51                print e
52            else:
53                self.fail('does not fail with wrong samplerate')
[6192ce7]54
[e1bfde5]55    def test_wrong_hop_size(self):
56        for p in list_of_sounds:
57            f = source(p, 0, -1)
58            print f.hop_size
59
60    def test_zero_hop_size(self):
61        for p in list_of_sounds:
62            f = source(p, 0, 0)
63            assert f.samplerate != 0
64            assert f.hop_size != 0
65            self.read_from_sink(f)
66
67if __name__ == '__main__':
68    from unittest import main
69    main()
Note: See TracBrowser for help on using the repository browser.