def test_track_basic():
# test data home None
track_default = guitarset.Track('03_BN3-119-G_solo')
assert track_default._data_home == os.path.join(DEFAULT_DATA_HOME,
'GuitarSet')
with pytest.raises(ValueError):
guitarset.Track('asdfasdf', data_home=TEST_DATA_HOME)
# test __repr__
assert isinstance(track_default.__repr__(), str)
def test_to_jams():
data_home = 'tests/resources/mir_datasets/GuitarSet'
track = guitarset.Track('03_BN3-119-G_solo', data_home=data_home)
jam = track.to_jams()
assert type(jam) == jams.JAMS
def test_track():
default_trackid = '03_BN3-119-G_solo'
data_home = TEST_DATA_HOME
track = guitarset.Track(default_trackid, data_home=data_home)
expected_attributes = {
'track_id': '03_BN3-119-G_solo',
'audio_hex_cln_path': 'tests/resources/mir_datasets/GuitarSet/' +
'audio_hex-pickup_debleeded/03_BN3-119-G_solo_hex_cln.wav',
'audio_hex_path': 'tests/resources/mir_datasets/GuitarSet/' +
'audio_hex-pickup_original/03_BN3-119-G_solo_hex.wav',
'audio_mic_path': 'tests/resources/mir_datasets/GuitarSet/' +
'audio_mono-mic/03_BN3-119-G_solo_mic.wav',
'audio_mix_path': 'tests/resources/mir_datasets/GuitarSet/' +
'audio_mono-pickup_mix/03_BN3-119-G_solo_mix.wav',
'jams_path': 'tests/resources/mir_datasets/GuitarSet/' +
'annotation/03_BN3-119-G_solo.jams',
'player_id': '03',
'tempo': 119,
'mode': 'solo',
'style': 'Bossa Nova',
}
expected_property_types = {
'beats': utils.BeatData,
'leadsheet_chords': utils.ChordData,
'inferred_chords': utils.ChordData,
'key_mode': utils.KeyData,
'pitch_contours': dict,
'notes': dict,
}
run_track_tests(track, expected_attributes, expected_property_types)
assert type(track.pitch_contours['E']) is utils.F0Data
assert type(track.notes['E']) is utils.NoteData
# -*- coding: utf-8 -*-
from __future__ import absolute_import
import numpy as np
import os
import pytest
import jams
from mirdata import guitarset, utils
from tests.test_utils import DEFAULT_DATA_HOME
from tests.test_download_utils import mock_downloader
TEST_DATA_HOME = 'tests/resources/mir_datasets/GuitarSet'
TRACK = guitarset.Track('03_BN3-119-G_solo', data_home=TEST_DATA_HOME)
def test_track_basic():
# test data home None
track_default = guitarset.Track('03_BN3-119-G_solo')
assert track_default._data_home == os.path.join(DEFAULT_DATA_HOME,
'GuitarSet')
with pytest.raises(ValueError):
guitarset.Track('asdfasdf', data_home=TEST_DATA_HOME)
# test __repr__
assert isinstance(track_default.__repr__(), str)
def test_track_simple_attributes():
