forked from librosa/librosa
/
test_onset.py
241 lines (172 loc) · 8.37 KB
/
test_onset.py
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
#!/usr/bin/env python
# CREATED:2013-03-11 18:14:30 by Brian McFee <brm2132@columbia.edu>
# unit tests for librosa.onset
from __future__ import print_function
import pytest
# Disable cache
import os
try:
os.environ.pop('LIBROSA_CACHE_DIR')
except:
pass
import warnings
import numpy as np
import librosa
__EXAMPLE_FILE = os.path.join('tests', 'data', 'test1_22050.wav')
def test_onset_strength_audio():
def __test(y, sr, feature, n_fft, hop_length, lag, max_size, detrend, center, aggregate):
oenv = librosa.onset.onset_strength(y=y, sr=sr,
S=None,
detrend=detrend,
center=center,
aggregate=aggregate,
feature=feature,
n_fft=n_fft,
hop_length=hop_length,
lag=lag,
max_size=max_size)
assert oenv.ndim == 1
S = librosa.feature.melspectrogram(y=y,
n_fft=n_fft,
hop_length=hop_length)
target_shape = S.shape[-1]
if not detrend:
assert np.all(oenv >= 0)
assert oenv.shape[-1] == target_shape
y, sr = librosa.load(__EXAMPLE_FILE)
for feature in [None,
librosa.feature.melspectrogram,
librosa.feature.chroma_stft]:
for n_fft in [512, 2048]:
for hop_length in [n_fft // 2, n_fft // 4]:
for lag in [0, 1, 2]:
for max_size in [0, 1, 2]:
for detrend in [False, True]:
for center in [False, True]:
for aggregate in [None, np.mean, np.max]:
if lag < 1 or max_size < 1:
tf = pytest.mark.xfail(__test, raises=librosa.ParameterError)
else:
tf = __test
yield (tf, y, sr, feature, n_fft,
hop_length, lag, max_size, detrend, center, aggregate)
tf = pytest.mark.xfail(__test, raises=librosa.ParameterError)
yield (tf, None, sr, feature, n_fft,
hop_length, lag, max_size, detrend, center, aggregate)
def test_onset_strength_spectrogram():
def __test(S, sr, feature, n_fft, hop_length, detrend, center):
oenv = librosa.onset.onset_strength(y=None, sr=sr,
S=S,
detrend=detrend,
center=center,
aggregate=aggregate,
feature=feature,
n_fft=n_fft,
hop_length=hop_length)
assert oenv.ndim == 1
target_shape = S.shape[-1]
if not detrend:
assert np.all(oenv >= 0)
assert oenv.shape[-1] == target_shape
y, sr = librosa.load(__EXAMPLE_FILE)
S = librosa.feature.melspectrogram(y=y, sr=sr)
for feature in [None,
librosa.feature.melspectrogram,
librosa.feature.chroma_stft]:
for n_fft in [512, 2048]:
for hop_length in [n_fft // 2, n_fft // 4]:
for detrend in [False, True]:
for center in [False, True]:
for aggregate in [None, np.mean, np.max]:
yield (__test, S, sr, feature, n_fft,
hop_length, detrend, center)
tf = pytest.mark.xfail(__test, raises=librosa.ParameterError)
yield (tf, None, sr, feature, n_fft,
hop_length, detrend, center)
def test_onset_strength_multi():
y, sr = librosa.load(__EXAMPLE_FILE)
S = librosa.feature.melspectrogram(y=y, sr=sr)
channels = np.linspace(0, S.shape[0], num=5).astype(int)
for lag in [1, 2, 3]:
for max_size in [1]:
# We only test with max_size=1 here to make the sub-band slicing test simple
odf_multi = librosa.onset.onset_strength_multi(S=S,
lag=lag, max_size=1,
channels=channels)
assert len(odf_multi) == len(channels) - 1
for i, (s, t) in enumerate(zip(channels, channels[1:])):
odf_single = librosa.onset.onset_strength(S=S[s:t],
lag=lag,
max_size=1)
assert np.allclose(odf_single, odf_multi[i])
def test_onset_detect_real():
def __test(y, sr, oenv, hop_length, bt):
onsets = librosa.onset.onset_detect(y=y, sr=sr, onset_envelope=oenv,
hop_length=hop_length,
backtrack=bt)
if bt:
assert np.all(onsets >= 0)
else:
assert np.all(onsets > 0)
assert np.all(onsets < len(y) * sr // hop_length)
if oenv is not None:
assert np.all(onsets < len(oenv))
y, sr = librosa.load(__EXAMPLE_FILE)
# Test with no signal
yield pytest.mark.xfail(__test, raises=librosa.ParameterError), None, sr, None, 512, False
for hop_length in [64, 512, 2048]:
oenv = librosa.onset.onset_strength(y=y, sr=sr, hop_length=hop_length)
for bt in [False, True]:
yield __test, y, sr, None, hop_length, bt
yield __test, y, sr, oenv, hop_length, bt
def test_onset_detect_const():
def __test(y, sr, oenv, hop_length):
onsets = librosa.onset.onset_detect(y=y, sr=sr, onset_envelope=oenv,
hop_length=hop_length)
assert len(onsets) == 0
sr = 22050
duration = 3.0
for f in [np.zeros, np.ones]:
y = f(int(duration * sr))
for hop_length in [64, 512, 2048]:
yield __test, y, sr, None, hop_length
yield __test, -y, sr, None, hop_length
oenv = librosa.onset.onset_strength(y=y,
sr=sr,
hop_length=hop_length)
yield __test, y, sr, oenv, hop_length
def test_onset_units():
def __test(units, hop_length, y, sr):
b1 = librosa.onset.onset_detect(y=y, sr=sr, hop_length=hop_length)
b2 = librosa.onset.onset_detect(y=y, sr=sr, hop_length=hop_length,
units=units)
t1 = librosa.frames_to_time(b1, sr=sr, hop_length=hop_length)
if units == 'time':
t2 = b2
elif units == 'samples':
t2 = librosa.samples_to_time(b2, sr=sr)
elif units == 'frames':
t2 = librosa.frames_to_time(b2, sr=sr, hop_length=hop_length)
assert np.allclose(t1, t2)
for sr in [None, 44100]:
y, sr = librosa.load(__EXAMPLE_FILE, sr=sr)
for hop_length in [512, 1024]:
for units in ['frames', 'time', 'samples']:
yield __test, units, hop_length, y, sr
yield pytest.mark.xfail(__test, raises=librosa.ParameterError), 'bad units', hop_length, y, sr
def test_onset_backtrack():
y, sr = librosa.load(__EXAMPLE_FILE)
oenv = librosa.onset.onset_strength(y=y, sr=sr)
onsets = librosa.onset.onset_detect(onset_envelope=oenv, backtrack=False)
def __test(energy):
# Test backtracking
onsets_bt = librosa.onset.onset_backtrack(onsets, energy)
# Make sure there are no negatives
assert np.all(onsets_bt >= 0)
# And that we never roll forward
assert np.all(onsets_bt <= onsets)
# And that the detected peaks are actually minima
assert np.all(energy[onsets_bt] <= energy[np.maximum(0, onsets_bt - 1)])
yield __test, oenv
rmse = librosa.feature.rmse(y=y)
yield __test, rmse