def test_track(): default_trackid = "some_id" data_home = "tests/resources/mir_datasets/dataset" dataset = example.Dataset(data_home) track = dataset.track(default_trackid) expected_attributes = { "track_id": "some_id", "audio_path": "tests/resources/mir_datasets/example/" + "Wavfile/some_id.wav", "song_id": "some_id", "annotation_path": "tests/resources/mir_datasets/example/annotation/some_id.pv", } expected_property_types = {"annotation": annotations.XData} assert track._track_paths == { "audio": ["Wavfile/some_id.wav", "278ae003cb0d323e99b9a643c0f2eeda"], "annotation": ["Annotation/some_id.pv", "0d93a011a9e668fd80673049089bbb14"], } run_track_tests(track, expected_attributes, expected_property_types) # test audio loading functions audio, sr = track.audio assert sr == 44100 assert audio.shape == (44100 * 2, )
def test_to_jams(): default_trackid = "some_id" data_home = "tests/resources/mir_datasets/dataset" dataset = example.Dataset(data_home) track = dataset.track(default_trackid) jam = track.to_jams() annotations = jam.search(namespace="annotation")[0]["data"] assert [annotation.time for annotation in annotations] == [0.027, 0.232] assert [annotation.duration for annotation in annotations] == [ 0.20500000000000002, 0.736, ]
def test_metadata(): data_home = "tests/resources/mir_datasets/dataset" dataset = example.Dataset(data_home) metadata = dataset._metadata assert metadata["some_id"] == "something"