JOJ3-config-generator/tests/create/utils.py
张泊明518370910136 4103b18673
All checks were successful
build / build (push) Successful in 1m58s
chore: better test failure hints
2025-03-18 04:21:28 -04:00

32 lines
1.3 KiB
Python

import difflib
import json
from pathlib import Path
import tomli
from joj3_config_generator.generator import create_joj3_task_conf
from joj3_config_generator.models import answer
def load_case(case_name: str) -> None:
root = Path(__file__).resolve().parent
answers_json_path = root / case_name / "answers.json"
task_toml_path = root / case_name / "task.toml"
answers_dict = json.loads(answers_json_path.read_text())
language = next(x for x in answer.LANGUAGES if str(x) == answers_dict["language"])
language.set_stages(answers_dict["stages"])
language.set_attribute(answers_dict["attribute"])
answers = answer.Answers(name=answers_dict["name"], language=language)
expected_result = tomli.loads(task_toml_path.read_text())
result = create_joj3_task_conf(answers).model_dump(
mode="json", by_alias=True, exclude_none=True, exclude_unset=True
)
if result != expected_result:
result_str = json.dumps(result, indent=2, ensure_ascii=False).splitlines()
expected_str = json.dumps(
expected_result, indent=2, ensure_ascii=False
).splitlines()
diff = "\n".join(difflib.ndiff(expected_str, result_str))
print(f"Test case '{case_name}' failed!\nDifferences:\n{diff}")
assert result == expected_result