JOJ/JOJ3-config-generator
27
1
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

fix: type annotation

Browse Source
This commit is contained in:
李衍志523370910113 2024-11-09 14:54:39 +08:00
parent 230b777e40
commit ec5c77782a
3 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
joj3_config_generator/lib/task.py
View File

@ -1,5 +1,5 @@
import shlex import shlex
from typing import Any, Dict, Tuple from typing import Tuple
import rtoml import rtoml
@ -13,7 +13,7 @@ from joj3_config_generator.models.result import Stage as ResultStage
from joj3_config_generator.models.task import Stage as TaskStage from joj3_config_generator.models.task import Stage as TaskStage
def remove_nulls(d: Dict[str, Any]) -> Dict[str, Any]: def remove_nulls(d: result.Config) -> result.Config:
if isinstance(d, dict): if isinstance(d, dict):
return {k: remove_nulls(v) for k, v in d.items() if v is not None} return {k: remove_nulls(v) for k, v in d.items() if v is not None}
elif isinstance(d, list): elif isinstance(d, list):

3
tests/convert/test_convert_cases.py
View File

@ -4,4 +4,5 @@ from tests.convert.utils import load_case
def test_basic() -> None: def test_basic() -> None:
load_case("basic") load_case("basic")
test_basic()
test_basic()

2
tests/convert/utils.py
View File

@ -29,5 +29,5 @@ def read_convert_files(
def load_case(case_name: str) -> None: def load_case(case_name: str) -> None:
repo, task, expected_result = read_convert_files(case_name) repo, task, expected_result = read_convert_files(case_name)
result = convert(repo, task).model_dump(mode="json", by_alias=True) result = convert(repo, task).model_dump(mode="json", by_alias=True)
assert result == expected_result assert result == expected_result