update one track test paths for tests

Natooz · Sep 28, 2023 · 01cdac0 · 01cdac0
1 parent 065fff2
commit 01cdac0
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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"),
     )