def test_track():
# test data home None
track_default = medley_solos_db.Track('d07b1fc0-567d-52c2-fef4-239f31c9d40e')
assert track_default._data_home == os.path.join(
DEFAULT_DATA_HOME, 'Medley-solos-DB'
)
data_home = 'tests/resources/mir_datasets/Medley-solos-DB'
with pytest.raises(ValueError):
medley_solos_db.Track('asdfasdf', data_home=data_home)
track = medley_solos_db.Track(
'd07b1fc0-567d-52c2-fef4-239f31c9d40e', data_home=data_home
)
assert track.track_id == 'd07b1fc0-567d-52c2-fef4-239f31c9d40e'
assert track._data_home == data_home
assert track._track_paths == {
'audio': [
'audio/Medley-solos-DB_validation-3_d07b1fc0-567d-52c2-fef4-239f31c9d40e.wav',
'53ed28731399b67425775be598b50d1c',
]
}
assert (
track.audio_path
== 'tests/resources/mir_datasets/Medley-solos-DB/audio/'
+ 'Medley-solos-DB_validation-3_d07b1fc0-567d-52c2-fef4-239f31c9d40e.wav'
)
assert track.instrument == 'flute'
assert track.instrument_id == 3
assert track.song_id == 210
assert track.subset == "validation"
y, sr = track.audio
assert y.shape == (65536,)
assert sr == 22050
repr_string = (
'Medley-solos-DB Track(track_id=d07b1fc0-567d-52c2-fef4-239f31c9d40e, '
+ 'audio_path=tests/resources/mir_datasets/Medley-solos-DB/audio/'
+ 'Medley-solos-DB_validation-3_d07b1fc0-567d-52c2-fef4-239f31c9d40e.wav, '
+ 'instrument=flute, song_id=210, subset=validation)'
)
assert track.__repr__() == repr_string
def test_to_jams():
data_home = 'tests/resources/mir_datasets/Medley-solos-DB'
track = medley_solos_db.Track('d07b1fc0-567d-52c2-fef4-239f31c9d40e',
data_home=data_home)
jam = track.to_jams()
assert jam['sandbox']['instrument'] == 'flute'
assert jam['sandbox']['instrument_id'] == 3
assert jam['sandbox']['song_id'] == 210
assert jam['sandbox']['subset'] == 'validation'
def test_track():
default_trackid = 'd07b1fc0-567d-52c2-fef4-239f31c9d40e'
data_home = 'tests/resources/mir_datasets/Medley-solos-DB'
track = medley_solos_db.Track(default_trackid, data_home=data_home)
expected_attributes = {
'track_id': 'd07b1fc0-567d-52c2-fef4-239f31c9d40e',
'audio_path': 'tests/resources/mir_datasets/Medley-solos-DB/' +
'audio/Medley-solos-DB_validation-3_d07b1fc0-567d-52c2-fef4-239f31c9d40e.wav',
'instrument': 'flute',
'instrument_id': 3,
'song_id': 210,
'subset': 'validation'
}
expected_property_types = {}
run_track_tests(track, expected_attributes, expected_property_types)
y, sr = track.audio
assert y.shape == (65536, )
assert sr == 22050