51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
|
import os
|
||
|
|
import shutil
|
||
|
|
import glob
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from conftest import convert
|
||
|
|
from tja2fumen.parsers import parse_fumen
|
||
|
|
|
||
|
|
CUSTOM_TJA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)),
|
||
|
|
"data", "custom_tjas")
|
||
|
|
# CUSTOM_TJA_DIR = os.path.join("D:\\", "games", "TaikoTDM",
|
||
|
|
# "CustomSongSources", "ESE")
|
||
|
|
CUSTOM_TJAS = sum([
|
||
|
|
# For each file in the subfolder, keep only if it ends in `.tja`
|
||
|
|
[(root, f) for f in files if f.endswith(".tja")]
|
||
|
|
# Iterate through all subfolders in the "unpaired_tjas" folder
|
||
|
|
for root, _, files in os.walk(CUSTOM_TJA_DIR)
|
||
|
|
], []) # sum([list of lists], []) -> https://stackoverflow.com/a/716489
|
||
|
|
CUSTOM_TJA_PATHS = [os.path.join(root, f) for (root, f) in CUSTOM_TJAS]
|
||
|
|
CUSTOM_TJA_IDS = [f for (root, f) in CUSTOM_TJAS]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('path_tja', CUSTOM_TJA_PATHS, ids=CUSTOM_TJA_IDS)
|
||
|
|
@pytest.mark.skipif("CI" in os.environ,
|
||
|
|
reason="Test is only for local debugging")
|
||
|
|
def test_converted_custom_tjas(path_tja, tmp_path, entry_point):
|
||
|
|
"""
|
||
|
|
A test purely to aid with debugging. It lets me drop a .tja into a
|
||
|
|
pre-determined folder and run the conversion, allowing me to set
|
||
|
|
breakpoints and debug internal state without any tedious setup.
|
||
|
|
"""
|
||
|
|
# Define the testing directory
|
||
|
|
path_test = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
|
||
|
|
# Copy input TJA to working directory
|
||
|
|
path_tja_tmp = str(tmp_path / "test.tja")
|
||
|
|
shutil.copy(path_tja, path_tja_tmp)
|
||
|
|
|
||
|
|
# Convert TJA file to fumen files
|
||
|
|
convert(path_test, path_tja_tmp, entry_point)
|
||
|
|
|
||
|
|
# Fetch output fumen paths
|
||
|
|
paths_out = glob.glob(os.path.join(tmp_path, "*.bin"))
|
||
|
|
assert paths_out, f"No bin files generated in {tmp_path}"
|
||
|
|
order = "xmhne" # Ura Oni -> Oni -> Hard -> Normal -> Easy
|
||
|
|
paths_out = sorted(paths_out,
|
||
|
|
key=lambda s: [order.index(c) if c in order
|
||
|
|
else len(order) for c in s])
|
||
|
|
for path_out in paths_out:
|
||
|
|
parse_fumen(path_out, exclude_empty_measures=False)
|