31 lines
1006 B
Python
31 lines
1006 B
Python
import json
|
|
from pathlib import Path
|
|
from typing import Any, Dict, Tuple
|
|
|
|
import rtoml
|
|
|
|
from joj3_config_generator.convert import convert
|
|
from joj3_config_generator.models import repo, task
|
|
|
|
|
|
def read_convert_files(
|
|
case_name: str,
|
|
) -> Tuple[repo.Config, task.Config, Dict[str, Any]]:
|
|
root = Path(__file__).resolve().parent
|
|
repo_toml_path = root / case_name / "repo.toml"
|
|
repo_toml = repo_toml_path.read_text() if repo_toml_path.exists() else ""
|
|
task_toml_path = root / case_name / "task.toml"
|
|
task_toml = task_toml_path.read_text() if task_toml_path.exists() else ""
|
|
result = json.loads((root / case_name / "task.json").read_text())
|
|
return (
|
|
repo.Config(**rtoml.loads(repo_toml)),
|
|
task.Config(**rtoml.loads(task_toml)),
|
|
result,
|
|
)
|
|
|
|
|
|
def load_case(case_name: str) -> None:
|
|
repo, task, expected_result = read_convert_files(case_name)
|
|
result = convert(repo, task).model_dump(mode="json", by_alias=True)
|
|
assert result == expected_result
|