source: python/tests/test_sink.py @ ff28d81

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

[tests] use run_module_suite so tests can run when pytest is not installed

  • Property mode set to 100755
File size: 3.8 KB
RevLine 
[e1bfde5]1#! /usr/bin/env python
2
[0b6d23d]3from numpy.testing import TestCase
[e1bfde5]4from aubio import fvec, source, sink
[ecd370b]5from utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path
6from _tools import parametrize, skipTest, assert_raises
[fa6373c]7
[e1bfde5]8list_of_sounds = list_all_sounds('sounds')
[2fe24df]9samplerates = [0, 44100, 8000, 32000]
10hop_sizes = [512, 1024, 64]
11
[e1bfde5]12path = None
13
[ab35262]14many_files = 300 # 256 opened files is too much
15
[2fe24df]16all_params = []
17for soundfile in list_of_sounds:
18    for hop_size in hop_sizes:
19        for samplerate in samplerates:
20            all_params.append((hop_size, samplerate, soundfile))
21
[8607a74]22class Test_aubio_sink(object):
[2fe24df]23
[fa6373c]24    def test_wrong_filename(self):
[ecd370b]25        with assert_raises(RuntimeError):
[fa6373c]26            sink('')
27
28    def test_wrong_samplerate(self):
[ecd370b]29        with assert_raises(RuntimeError):
[fa6373c]30            sink(get_tmp_sink_path(), -1)
31
32    def test_wrong_samplerate_too_large(self):
[ecd370b]33        with assert_raises(RuntimeError):
[fa6373c]34            sink(get_tmp_sink_path(), 1536001, 2)
35
36    def test_wrong_channels(self):
[ecd370b]37        with assert_raises(RuntimeError):
[fa6373c]38            sink(get_tmp_sink_path(), 44100, -1)
39
40    def test_wrong_channels_too_large(self):
[ecd370b]41        with assert_raises(RuntimeError):
[fa6373c]42            sink(get_tmp_sink_path(), 44100, 202020)
43
[9e6695d]44    def test_many_sinks(self):
[ab35262]45        from tempfile import mkdtemp
46        import os.path
47        import shutil
48        tmpdir = mkdtemp()
49        sink_list = []
50        for i in range(many_files):
51            path = os.path.join(tmpdir, 'f-' + str(i) + '.wav')
52            g = sink(path, 0)
53            sink_list.append(g)
54            write = 32
[0b6d23d]55            for _ in range(200):
[9e6695d]56                vec = fvec(write)
57                g(vec, write)
[ab35262]58            g.close()
59        shutil.rmtree(tmpdir)
60
[ecd370b]61    @parametrize('hop_size, samplerate, path', all_params)
[2fe24df]62    def test_read_and_write(self, hop_size, samplerate, path):
63        try:
64            f = source(path, samplerate, hop_size)
65        except RuntimeError as e:
[ecd370b]66            err_msg = '{:s} (hop_s = {:d}, samplerate = {:d})'
67            skipTest(err_msg.format(str(e), hop_size, samplerate))
[2fe24df]68        if samplerate == 0: samplerate = f.samplerate
69        sink_path = get_tmp_sink_path()
70        g = sink(sink_path, samplerate)
71        total_frames = 0
72        while True:
73            vec, read = f()
74            g(vec, read)
75            total_frames += read
76            if read < f.hop_size: break
77        del_tmp_sink_path(sink_path)
[060a3135]78
[ecd370b]79    @parametrize('hop_size, samplerate, path', all_params)
[2fe24df]80    def test_read_and_write_multi(self, hop_size, samplerate, path):
81        try:
82            f = source(path, samplerate, hop_size)
83        except RuntimeError as e:
[ecd370b]84            err_msg = '{:s} (hop_s = {:d}, samplerate = {:d})'
85            skipTest(err_msg.format(str(e), hop_size, samplerate))
[2fe24df]86        if samplerate == 0: samplerate = f.samplerate
87        sink_path = get_tmp_sink_path()
88        g = sink(sink_path, samplerate, channels = f.channels)
89        total_frames = 0
90        while True:
91            vec, read = f.do_multi()
92            g.do_multi(vec, read)
93            total_frames += read
94            if read < f.hop_size: break
95        del_tmp_sink_path(sink_path)
[060a3135]96
[4949182]97    def test_close_file(self):
98        samplerate = 44100
[ab35262]99        sink_path = get_tmp_sink_path()
100        g = sink(sink_path, samplerate)
[4949182]101        g.close()
[ab35262]102        del_tmp_sink_path(sink_path)
[4949182]103
104    def test_close_file_twice(self):
105        samplerate = 44100
[ab35262]106        sink_path = get_tmp_sink_path()
107        g = sink(sink_path, samplerate)
[4949182]108        g.close()
109        g.close()
[ab35262]110        del_tmp_sink_path(sink_path)
[4949182]111
[81f738e]112    def test_read_with(self):
[6448d31]113        samplerate = 44100
114        sink_path = get_tmp_sink_path()
[81f738e]115        vec = fvec(128)
116        with sink(sink_path, samplerate) as g:
[6448d31]117            for _ in range(10):
[81f738e]118                g(vec, 128)
119
[e1bfde5]120if __name__ == '__main__':
[7fd92ca]121    from _tools import run_module_suite
122    run_module_suite()
Note: See TracBrowser for help on using the repository browser.