3 from numpy.testing import TestCase, assert_equal, assert_almost_equal
4 from aubio import fvec, source, sink
5 from numpy import array
6 from utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path
8 list_of_sounds = list_all_sounds('sounds')
11 many_files = 300 # 256 opened files is too much
13 class aubio_sink_test_case(TestCase):
15 def test_many_sinks(self):
16 from tempfile import mkdtemp
21 for i in range(many_files):
22 path = os.path.join(tmpdir, 'f-' + str(i) + '.wav')
32 def test_many_sinks_not_closed(self):
33 from tempfile import mkdtemp
39 for i in range(many_files):
40 path = os.path.join(tmpdir, 'f-' + str(i) + '.wav')
50 self.fail("does not fail on too many files open")
55 def test_read_and_write(self):
57 if not len(list_of_sounds):
58 self.skipTest('add some sound files in \'python/tests/sounds\'')
60 for path in list_of_sounds:
61 for samplerate, hop_size in zip([0, 44100, 8000, 32000], [512, 1024, 64, 256]):
62 f = source(path, samplerate, hop_size)
63 if samplerate == 0: samplerate = f.samplerate
64 sink_path = get_tmp_sink_path()
65 g = sink(sink_path, samplerate)
71 if read < f.hop_size: break
73 print ("read", "%.2fs" % (total_frames / float(f.samplerate) ) ),
74 print ("(", total_frames, "frames", "in" ),
75 print (total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")" ),
76 print ("from", f.uri ),
78 del_tmp_sink_path(sink_path)
80 def test_read_and_write_multi(self):
82 if not len(list_of_sounds):
83 self.skipTest('add some sound files in \'python/tests/sounds\'')
85 for path in list_of_sounds:
86 for samplerate, hop_size in zip([0, 44100, 8000, 32000], [512, 1024, 64, 256]):
87 f = source(path, samplerate, hop_size)
88 if samplerate == 0: samplerate = f.samplerate
89 sink_path = get_tmp_sink_path()
90 g = sink(sink_path, samplerate, channels = f.channels)
93 vec, read = f.do_multi()
96 if read < f.hop_size: break
98 print ("read", "%.2fs" % (total_frames / float(f.samplerate) ) ),
99 print ("(", total_frames, "frames", "in" ),
100 print (f.channels, "channels", "in" ),
101 print (total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")" ),
102 print ("from", f.uri ),
103 print ("to", g.uri ),
104 print ("in", g.channels, "channels")
105 del_tmp_sink_path(sink_path)
107 def test_close_file(self):
109 sink_path = get_tmp_sink_path()
110 g = sink(sink_path, samplerate)
112 del_tmp_sink_path(sink_path)
114 def test_close_file_twice(self):
116 sink_path = get_tmp_sink_path()
117 g = sink(sink_path, samplerate)
120 del_tmp_sink_path(sink_path)
122 if __name__ == '__main__':
123 from unittest import main