def test_track():
# test data home None
track_default = salami.Track('2')
assert track_default._data_home == os.path.join(DEFAULT_DATA_HOME,
'Salami')
# test specific data home
data_home = 'tests/resources/mir_datasets/Salami'
with pytest.raises(ValueError):
salami.Track('asdfasdf', data_home=data_home)
track = salami.Track('2', data_home=data_home)
# test attributes are loaded as expected
assert track.track_id == '2'
assert track._data_home == data_home
assert track._track_paths == {
'audio': ['audio/2.mp3', '76789a17bda0dd4d1d7e77424099c814'],
'annotator_1_uppercase': [
'salami-data-public-master/annotations/2/parsed/textfile1_uppercase.txt',
'54ba0804f720d85d195dcd7ffaec0794',
],
'annotator_1_lowercase': [
'salami-data-public-master/annotations/2/parsed/textfile1_lowercase.txt',
'30ff127ff68c61039b94a44ab6ddda34',
],
'annotator_2_uppercase': [
'salami-data-public-master/annotations/2/parsed/textfile2_uppercase.txt',
'e9dca8577f028d3505ff1e5801397b2f',
],
'annotator_2_lowercase': [
'salami-data-public-master/annotations/2/parsed/textfile2_lowercase.txt',
'546a783c7b8bf96f2d718c7a4f114699',
],
}
assert track.audio_path == 'tests/resources/mir_datasets/Salami/' + 'audio/2.mp3'
assert track.source == 'Codaich'
assert track.annotator_1_id == '5'
assert track.annotator_2_id == '8'
assert track.duration_sec == '264'
assert track.title == 'For_God_And_Country'
assert track.artist == 'The_Smashing_Pumpkins'
assert track.annotator_1_time == '37'
assert track.annotator_2_time == '45'
assert track.broad_genre == 'popular'
assert track.genre == 'Alternative_Pop___Rock'
# test that cached properties don't fail and have the expected type
assert type(track.sections_annotator_1_uppercase) is utils.SectionData
assert type(track.sections_annotator_1_lowercase) is utils.SectionData
assert type(track.sections_annotator_2_uppercase) is utils.SectionData
assert type(track.sections_annotator_2_lowercase) is utils.SectionData
# # test audio loading functions
# y, sr = track.audio
# assert sr == 44100
# assert y.shape == (89856, )
repr_string = (
"Salami Track(track_id=2, " +
"audio_path=tests/resources/mir_datasets/Salami/audio/2.mp3, " +
"source=Codaich, title=For_God_And_Country, " +
"artist=The_Smashing_Pumpkins, duration_sec=264, annotator_1_id=5, " +
"annotator_2_id=8, annotator_1_time=37, annotator_2_time=45, " +
"broad_genre=popular, genre=Alternative_Pop___Rock, " +
"sections_annotator_1_uppercase=SectionData('start_times', 'end_times', 'sections'), "
+
"sections_annotator_1_lowercase=SectionData('start_times', 'end_times', 'sections'), "
+
"sections_annotator_2_uppercase=SectionData('start_times', 'end_times', 'sections'), "
+
"sections_annotator_2_lowercase=SectionData('start_times', 'end_times', 'sections')"
)
assert track.__repr__() == repr_string
# Test file with missing annotations
track = salami.Track('192', data_home=data_home)
# test attributes
assert track.source == 'Codaich'
assert track.annotator_1_id == '16'
assert track.annotator_2_id == '14'
assert track.duration_sec == '209'
assert track.title == 'Sull__aria'
assert track.artist == 'Compilations'
assert track.annotator_1_time == '20'
assert track.annotator_2_time == ''
assert track.broad_genre == 'classical'
assert track.genre == 'Classical_-_Classical'
assert track.track_id == '192'
assert track._data_home == data_home
assert track._track_paths == {
'audio': ['audio/192.mp3', 'd954d5dc9f17d66155d3310d838756b8'],
'annotator_1_uppercase': [
'salami-data-public-master/annotations/192/parsed/textfile1_uppercase.txt',
'4d268cfd27fe011dbe579f25f8d125ce',
],
'annotator_1_lowercase': [
'salami-data-public-master/annotations/192/parsed/textfile1_lowercase.txt',
'6640237e7844d0d9d37bf21cf96a2690',
],
'annotator_2_uppercase': [None, None],
'annotator_2_lowercase': [None, None],
}
# test that cached properties don't fail and have the expected type
assert type(track.sections_annotator_1_uppercase) is utils.SectionData
assert type(track.sections_annotator_1_lowercase) is utils.SectionData
assert track.sections_annotator_2_uppercase is None
assert track.sections_annotator_2_lowercase is None
# Test file with missing annotations
track = salami.Track('1015', data_home=data_home)
assert track._track_paths == {
'audio': ['audio/1015.mp3', '811a4a6b46f0c15a61bfb299b21ebdc4'],
'annotator_1_uppercase': [None, None],
'annotator_1_lowercase': [None, None],
'annotator_2_uppercase': [
'salami-data-public-master/annotations/1015/parsed/textfile2_uppercase.txt',
'e4a268342a45fdffd8ec9c3b8287ad8b',
],
'annotator_2_lowercase': [
'salami-data-public-master/annotations/1015/parsed/textfile2_lowercase.txt',
'201642fcea4a27c60f7b48de46a82234',
],
}
# test that cached properties don't fail and have the expected type
assert track.sections_annotator_1_uppercase is None
assert track.sections_annotator_1_lowercase is None
assert type(track.sections_annotator_2_uppercase) is utils.SectionData
assert type(track.sections_annotator_2_lowercase) is utils.SectionData
def test_to_jams():
data_home = 'tests/resources/mir_datasets/Salami'
track = salami.Track('2', data_home=data_home)
jam = track.to_jams()
segments = jam.search(namespace='multi_segment')[0]['data']
assert [segment.time for segment in segments] == [
0.0,
0.0,
0.464399092,
0.464399092,
5.191269841,
14.379863945,
254.821632653,
258.900453514,
263.205419501,
263.205419501,
]
assert [segment.duration for segment in segments] == [
0.464399092,
0.464399092,
13.915464853,
4.726870749000001,
249.630362812,
248.82555555599998,
4.078820860999997,
4.304965987000003,
1.6797959180000248,
1.6797959180000248,
]
assert [segment.value for segment in segments] == [
{
'label': 'Silence',
'level': 0
},
{
'label': 'Silence',
'level': 1
},
{
'label': 'A',
'level': 0
},
{
'label': 'b',
'level': 1
},
{
'label': 'b',
'level': 1
},
{
'label': 'B',
'level': 0
},
{
'label': 'ab',
'level': 1
},
{
'label': 'ab',
'level': 1
},
{
'label': 'Silence',
'level': 0
},
{
'label': 'Silence',
'level': 1
},
]
assert [segment.confidence for segment in segments] == [
None,
None,
None,
None,
None,
None,
None,
None,
None,
None,
]
assert jam['file_metadata']['title'] == 'For_God_And_Country'
assert jam['file_metadata']['artist'] == 'The_Smashing_Pumpkins'
def test_load_track_wrong_metadata():
with pytest.raises(OSError):
salami.METADATA = {'data_home': 'test'}
salami.Track('10', 'wrong_data_home')
def test_load_track_no_metadata():
with pytest.raises(OSError):
salami.METADATA = None
salami.Track('10')
def test_load_track_invalid_track_id():
with pytest.raises(ValueError):
salami.Track('a_fake_track')
def test_track():
default_trackid = '2'
data_home = 'tests/resources/mir_datasets/Salami'
track = salami.Track(default_trackid, data_home=data_home)
expected_attributes = {
'track_id':
'2',
'audio_path':
'tests/resources/mir_datasets/Salami/' + 'audio/2.mp3',
'sections_annotator1_uppercase_path':
'tests/resources/mir_datasets/Salami/' +
'salami-data-public-hierarchy-corrections/annotations/2/parsed/textfile1_uppercase.txt',
'sections_annotator1_lowercase_path':
'tests/resources/mir_datasets/Salami/' +
'salami-data-public-hierarchy-corrections/annotations/2/parsed/textfile1_lowercase.txt',
'sections_annotator2_uppercase_path':
'tests/resources/mir_datasets/Salami/' +
'salami-data-public-hierarchy-corrections/annotations/2/parsed/textfile2_uppercase.txt',
'sections_annotator2_lowercase_path':
'tests/resources/mir_datasets/Salami/' +
'salami-data-public-hierarchy-corrections/annotations/2/parsed/textfile2_lowercase.txt',
'source':
'Codaich',
'annotator_1_id':
'5',
'annotator_2_id':
'8',
'duration':
264,
'title':
'For_God_And_Country',
'artist':
'The_Smashing_Pumpkins',
'annotator_1_time':
'37',
'annotator_2_time':
'45',
'broad_genre':
'popular',
'genre':
'Alternative_Pop___Rock',
}
expected_property_types = {
'sections_annotator_1_uppercase': utils.SectionData,
'sections_annotator_1_lowercase': utils.SectionData,
'sections_annotator_2_uppercase': utils.SectionData,
'sections_annotator_2_lowercase': utils.SectionData,
}
run_track_tests(track, expected_attributes, expected_property_types)
# test audio loading functions
y, sr = track.audio
assert sr == 44100
assert y.shape == (89856, )
# Test file with missing annotations
track = salami.Track('192', data_home=data_home)
# test attributes
assert track.source == 'Codaich'
assert track.annotator_1_id == '16'
assert track.annotator_2_id == '14'
assert track.duration == 209
assert track.title == 'Sull__aria'
assert track.artist == 'Compilations'
assert track.annotator_1_time == '20'
assert track.annotator_2_time == ''
assert track.broad_genre == 'classical'
assert track.genre == 'Classical_-_Classical'
assert track.track_id == '192'
assert track._data_home == data_home
assert track._track_paths == {
'audio': ['audio/192.mp3', 'd954d5dc9f17d66155d3310d838756b8'],
'annotator_1_uppercase': [
'salami-data-public-hierarchy-corrections/annotations/192/parsed/textfile1_uppercase.txt',
'4d268cfd27fe011dbe579f25f8d125ce',
],
'annotator_1_lowercase': [
'salami-data-public-hierarchy-corrections/annotations/192/parsed/textfile1_lowercase.txt',
'6640237e7844d0d9d37bf21cf96a2690',
],
'annotator_2_uppercase': [None, None],
'annotator_2_lowercase': [None, None],
}
# test that cached properties don't fail and have the expected type
assert type(track.sections_annotator_1_uppercase) is utils.SectionData
assert type(track.sections_annotator_1_lowercase) is utils.SectionData
assert track.sections_annotator_2_uppercase is None
assert track.sections_annotator_2_lowercase is None
# Test file with missing annotations
track = salami.Track('1015', data_home=data_home)
assert track._track_paths == {
'audio': ['audio/1015.mp3', '811a4a6b46f0c15a61bfb299b21ebdc4'],
'annotator_1_uppercase': [None, None],
'annotator_1_lowercase': [None, None],
'annotator_2_uppercase': [
'salami-data-public-hierarchy-corrections/annotations/1015/parsed/textfile2_uppercase.txt',
'e4a268342a45fdffd8ec9c3b8287ad8b',
],
'annotator_2_lowercase': [
'salami-data-public-hierarchy-corrections/annotations/1015/parsed/textfile2_lowercase.txt',
'201642fcea4a27c60f7b48de46a82234',
],
}
# test that cached properties don't fail and have the expected type
assert track.sections_annotator_1_uppercase is None
assert track.sections_annotator_1_lowercase is None
assert type(track.sections_annotator_2_uppercase) is utils.SectionData
assert type(track.sections_annotator_2_lowercase) is utils.SectionData
