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():