ba05d06cae8ca03ae80c40f72194f9d30e033d25
[aubio.git] / python / tests / test_source.py
1 #! /usr/bin/env python
2
3 from numpy.testing import TestCase, assert_equal, assert_almost_equal
4 from aubio import fvec, source
5 from numpy import array
6 from utils import list_all_sounds
7
8 list_of_sounds = list_all_sounds('sounds')
9 path = None
10
11 class aubio_source_test_case_base(TestCase):
12
13     def setUp(self):
14         if not len(list_of_sounds): self.skipTest('add some sound files in \'python/tests/sounds\'')
15
16 class aubio_source_test_case(aubio_source_test_case_base):
17
18     def test_close_file(self):
19         samplerate = 0 # use native samplerate
20         hop_size = 256
21         for p in list_of_sounds:
22             f = source(p, samplerate, hop_size)
23             f.close()
24
25     def test_close_file_twice(self):
26         samplerate = 0 # use native samplerate
27         hop_size = 256
28         for p in list_of_sounds:
29             f = source(p, samplerate, hop_size)
30             f.close()
31             f.close()
32
33 class aubio_source_read_test_case(aubio_source_test_case_base):
34
35     def read_from_sink(self, f):
36         total_frames = 0
37         while True:
38             vec, read = f()
39             total_frames += read
40             if read < f.hop_size: break
41         print "read", "%.2fs" % (total_frames / float(f.samplerate) ),
42         print "(", total_frames, "frames", "in",
43         print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")",
44         print "from", f.uri
45
46     def test_samplerate_hopsize(self):
47         for p in list_of_sounds:
48             for samplerate, hop_size in zip([0, 44100, 8000, 32000], [ 512, 512, 64, 256]):
49                 f = source(p, samplerate, hop_size)
50                 assert f.samplerate != 0
51                 self.read_from_sink(f)
52
53     def test_samplerate_none(self):
54         for p in list_of_sounds:
55             f = source(p)
56             assert f.samplerate != 0
57             self.read_from_sink(f)
58
59     def test_samplerate_0(self):
60         for p in list_of_sounds:
61             f = source(p, 0)
62             assert f.samplerate != 0
63             self.read_from_sink(f)
64
65     def test_wrong_samplerate(self):
66         for p in list_of_sounds:
67             try:
68                 f = source(p, -1)
69             except ValueError, e:
70                 pass
71             else:
72                 self.fail('negative samplerate does not raise ValueError')
73
74     def test_wrong_hop_size(self):
75         for p in list_of_sounds:
76             try:
77                 f = source(p, 0, -1)
78             except ValueError, e:
79                 pass
80             else:
81                 self.fail('negative hop_size does not raise ValueError')
82
83     def test_zero_hop_size(self):
84         for p in list_of_sounds:
85             f = source(p, 0, 0)
86             assert f.samplerate != 0
87             assert f.hop_size != 0
88             self.read_from_sink(f)
89
90 class aubio_source_readmulti_test_case(aubio_source_read_test_case):
91
92     def read_from_sink(self, f):
93         total_frames = 0
94         while True:
95             vec, read = f.do_multi()
96             total_frames += read
97             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 and",
101         print total_frames / f.hop_size, "blocks", "at", "%dHz" % f.samplerate, ")",
102         print "from", f.uri
103
104 if __name__ == '__main__':
105     from unittest import main
106     main()