source: python/tests/test_sink.py @ 0229e51

feature/autosinkfeature/constantqfeature/pitchshiftfeature/pydocstringsfeature/timestretchpitchshiftsamplertimestretchyinfft+
Last change on this file since 0229e51 was 0229e51, checked in by Paul Brossier <piem@piem.org>, 4 years ago

python/tests/test_sink.py: remove useless many_sinks_not_closed and more cruft

  • Property mode set to 100755
File size: 2.8 KB
Line 
1#! /usr/bin/env python
2
3from numpy.testing import TestCase, assert_equal, assert_almost_equal
4from aubio import fvec, source, sink
5from numpy import array
6from utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path
7
8list_of_sounds = list_all_sounds('sounds')
9path = None
10
11many_files = 300 # 256 opened files is too much
12
13class aubio_sink_test_case(TestCase):
14
15    def test_many_sinks(self):
16        from tempfile import mkdtemp
17        import os.path
18        import shutil
19        tmpdir = mkdtemp()
20        sink_list = []
21        for i in range(many_files):
22            path = os.path.join(tmpdir, 'f-' + str(i) + '.wav')
23            g = sink(path, 0)
24            sink_list.append(g)
25            write = 32
26            for n in range(200):
27                vec = fvec(write)
28                g(vec, write)
29            g.close()
30        shutil.rmtree(tmpdir)
31
32    def test_read_and_write(self):
33
34        if not len(list_of_sounds):
35            self.skipTest('add some sound files in \'python/tests/sounds\'')
36
37        for path in list_of_sounds:
38            for samplerate, hop_size in zip([0, 44100, 8000, 32000], [512, 1024, 64, 256]):
39                f = source(path, samplerate, hop_size)
40                if samplerate == 0: samplerate = f.samplerate
41                sink_path = get_tmp_sink_path()
42                g = sink(sink_path, samplerate)
43                total_frames = 0
44                while True:
45                    vec, read = f()
46                    g(vec, read)
47                    total_frames += read
48                    if read < f.hop_size: break
49                del_tmp_sink_path(sink_path)
50
51    def test_read_and_write_multi(self):
52
53        if not len(list_of_sounds):
54            self.skipTest('add some sound files in \'python/tests/sounds\'')
55
56        for path in list_of_sounds:
57            for samplerate, hop_size in zip([0, 44100, 8000, 32000], [512, 1024, 64, 256]):
58                f = source(path, samplerate, hop_size)
59                if samplerate == 0: samplerate = f.samplerate
60                sink_path = get_tmp_sink_path()
61                g = sink(sink_path, samplerate, channels = f.channels)
62                total_frames = 0
63                while True:
64                    vec, read = f.do_multi()
65                    g.do_multi(vec, read)
66                    total_frames += read
67                    if read < f.hop_size: break
68                del_tmp_sink_path(sink_path)
69
70    def test_close_file(self):
71        samplerate = 44100
72        sink_path = get_tmp_sink_path()
73        g = sink(sink_path, samplerate)
74        g.close()
75        del_tmp_sink_path(sink_path)
76
77    def test_close_file_twice(self):
78        samplerate = 44100
79        sink_path = get_tmp_sink_path()
80        g = sink(sink_path, samplerate)
81        g.close()
82        g.close()
83        del_tmp_sink_path(sink_path)
84
85if __name__ == '__main__':
86    from unittest import main
87    main()
Note: See TracBrowser for help on using the repository browser.