diff --git a/tests/test_bpe.py b/tests/test_bpe.py index 1ad09da1..74e6f702 100644 --- a/tests/test_bpe.py +++ b/tests/test_bpe.py @@ -30,7 +30,7 @@ def test_bpe_conversion( - data_path: Union[str, Path, PurePath] = "./tests/Maestro_MIDIs" + data_path: Union[str, Path, PurePath] = "./tests/One_track_MIDIs" ): r"""Reads a few MIDI files, convert them into token sequences, convert them back to MIDI files. The converted back MIDI files should identical to original one, expect with note starting and ending diff --git a/tests/test_methods.py b/tests/test_methods.py index 7a45aad7..914f66a3 100644 --- a/tests/test_methods.py +++ b/tests/test_methods.py @@ -38,7 +38,7 @@ def test_convert_tensors(): """def time_data_augmentation_tokens_vs_mid(): from time import time tokenizers = [miditok.TSD(), miditok.REMI()] - data_paths = [Path('./tests/Maestro_MIDIs'), Path('./tests/Multitrack_MIDIs')] + data_paths = [Path('./tests/One_track_MIDIs'), Path('./tests/Multitrack_MIDIs')] for data_path in data_paths: for tokenizer in tokenizers: diff --git a/tests/test_pytorch_data_loading.py b/tests/test_pytorch_data_loading.py index b796f53b..4716578b 100644 --- a/tests/test_pytorch_data_loading.py +++ b/tests/test_pytorch_data_loading.py @@ -27,7 +27,7 @@ def test_dataset_ram(): multitrack_midis_paths = list(Path("tests", "Multitrack_MIDIs").glob("**/*.mid"))[ :3 ] - one_track_midis_paths = list(Path("tests", "Maestro_MIDIs").glob("**/*.mid"))[:3] + one_track_midis_paths = list(Path("tests", "One_track_MIDIs").glob("**/*.mid"))[:3] tokens_os_dir = Path("tests", "multitrack_tokens_os") # MIDI + One token stream diff --git a/tests/test_utils.py b/tests/test_utils.py index 9f373ac4..b485dfc3 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -20,7 +20,7 @@ def test_merge_tracks(): - midi = MidiFile(Path("tests", "Maestro_MIDIs", "Maestro_1.mid")) + midi = MidiFile(Path("tests", "One_track_MIDIs", "Maestro_1.mid")) original_track = deepcopy(midi.instruments[0]) midi.instruments.append(deepcopy(midi.instruments[0])) merge_tracks(midi.instruments) @@ -68,7 +68,7 @@ def test_merge_same_program_tracks_and_by_class(): def test_nb_pos(): tokenizer = REMI() _ = nb_bar_pos( - tokenizer(Path("tests", "Maestro_MIDIs", "Maestro_1.mid"))[0].ids, + tokenizer(Path("tests", "One_track_MIDIs", "Maestro_1.mid"))[0].ids, tokenizer["Bar_None"], tokenizer.token_ids_of_type("Position"), )