1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#! /usr/bin/env python
from numpy.testing import TestCase, assert_equal, assert_almost_equal
from aubio import onset, fvec
class aubio_onset_default(TestCase):
def test_members(self):
o = onset()
assert_equal ([o.buf_size, o.hop_size, o.method, o.samplerate],
[1024,512,'default',44100])
class aubio_onset_params(TestCase):
samplerate = 44100
def setUp(self):
self.o = onset(samplerate = self.samplerate)
def test_get_delay(self):
self.assertGreater(self.o.get_delay(), 0)
def test_get_delay_s(self):
self.assertGreater(self.o.get_delay_s(), 0.)
def test_get_delay_ms(self):
self.assertGreater(self.o.get_delay_ms(), 0.)
def test_get_minioi(self):
self.assertGreater(self.o.get_minioi(), 0)
def test_get_minioi_s(self):
self.assertGreater(self.o.get_minioi_s(), 0.)
def test_get_minioi_ms(self):
self.assertGreater(self.o.get_minioi_ms(), 0.)
def test_get_threshold(self):
self.assertGreater(self.o.get_threshold(), 0.)
def test_set_delay(self):
val = 256
self.o.set_delay(val)
assert_equal (self.o.get_delay(), val)
def test_set_delay_s(self):
val = .05
self.o.set_delay_s(val)
assert_almost_equal (self.o.get_delay_s(), val)
def test_set_delay_ms(self):
val = 50.
self.o.set_delay_ms(val)
assert_almost_equal (self.o.get_delay_ms(), val)
def test_set_minioi(self):
val = 200
self.o.set_minioi(val)
assert_equal (self.o.get_minioi(), val)
def test_set_minioi_s(self):
val = 0.04
self.o.set_minioi_s(val)
assert_almost_equal (self.o.get_minioi_s(), val)
def test_set_minioi_ms(self):
val = 40.
self.o.set_minioi_ms(val)
assert_almost_equal (self.o.get_minioi_ms(), val)
def test_set_threshold(self):
val = 0.2
self.o.set_threshold(val)
assert_almost_equal (self.o.get_threshold(), val)
class aubio_onset_96000(aubio_onset_params):
samplerate = 96000
class aubio_onset_32000(aubio_onset_params):
samplerate = 32000
class aubio_onset_8000(aubio_onset_params):
samplerate = 8000
class aubio_onset_coverate(TestCase):
# extra tests to execute the C routines and improve coverage
def test_all_methods(self):
for method in ['default', 'energy', 'hfc', 'complexdomain', 'complex',
'phase', 'wphase', 'mkl', 'kl', 'specflux', 'specdiff',
'old_default']:
o = onset(method=method, buf_size=512, hop_size=256)
o(fvec(256))
def test_get_methods(self):
o = onset(method='default', buf_size=512, hop_size=256)
assert o.get_silence() == -70
o.set_silence(-20)
assert_almost_equal(o.get_silence(), -20)
assert o.get_compression() == 1
o.set_compression(.99)
assert_almost_equal(o.get_compression(), .99)
assert o.get_awhitening() == 0
o.set_awhitening(1)
assert o.get_awhitening() == 1
o.get_last()
o.get_last_ms()
o.get_last_s()
o.get_descriptor()
o.get_thresholded_descriptor()
if __name__ == '__main__':
from unittest import main
main()
|
