def test_load_notes():
notes_path = (
'tests/resources/mir_datasets/DALI/annotations/'
+ '4b196e6c99574dd49ad00d56e132712b.gz'
)
note_data = dali.load_annotations_granularity(notes_path, 'notes')
assert type(note_data) == utils.NoteData
assert type(note_data.intervals) == np.ndarray
assert type(note_data.notes) == np.ndarray
assert np.array_equal(note_data.intervals[:, 0], np.array([24.125, 24.273, 24.420]))
assert np.array_equal(note_data.intervals[:, 1], np.array([24.273, 24.420, 24.568]))
assert np.array_equal(note_data.notes, np.array([1108.731, 1108.731, 1108.731]))
# load a file which doesn't exist
notes_none = dali.load_annotations_granularity('fake/file/path', 'notes')
assert notes_none is None
def test_load_paragraphs():
data_path = ('tests/resources/mir_datasets/DALI/annotations/' +
'4b196e6c99574dd49ad00d56e132712b.gz')
par_data = dali.load_annotations_granularity(data_path, 'paragraphs')
assert type(par_data) == utils.LyricData
assert type(par_data.start_times) == np.ndarray
assert type(par_data.end_times) == np.ndarray
assert type(par_data.lyrics) == np.ndarray
assert np.array_equal(par_data.start_times, np.array([24.125, 24.420]))
assert np.array_equal(par_data.end_times, np.array([24.420, 24.568]))
assert np.array_equal(par_data.lyrics, np.array(['why do', 'they']))
# load a file which doesn't exist
pars_none = dali.load_annotations_granularity('fake/file/path',
'paragraphs')
assert pars_none is None
def test_load_words():
data_path = ('tests/resources/mir_datasets/DALI/annotations/' +
'4b196e6c99574dd49ad00d56e132712b.gz')
word_data = dali.load_annotations_granularity(data_path, 'words')
assert type(word_data) == utils.LyricData
assert type(word_data.start_times) == np.ndarray
assert type(word_data.end_times) == np.ndarray
assert type(word_data.lyrics) == np.ndarray
assert np.array_equal(word_data.start_times,
np.array([24.125, 24.273, 24.42]))
assert np.array_equal(word_data.end_times,
np.array([24.273, 24.42, 24.568]))
assert np.array_equal(word_data.lyrics, np.array(['why', 'do', 'they']))
