32a241c28f
Some checks failed
nightly / Test against latest dependencies (py3.10) (push) Has been cancelled
nightly / Test against latest dependencies (py3.13) (push) Has been cancelled
tests / tests (3.13, locked) (push) Has been cancelled
tests / tests (3.13, unlocked) (push) Has been cancelled
tests / pyright (3.10) (push) Has been cancelled
tests / lint-format (push) Has been cancelled
tests / tests (3.10, locked) (push) Has been cancelled
tests / tests (3.11, locked) (push) Has been cancelled
tests / tests (3.12, locked) (push) Has been cancelled
tests / pyright (3.11) (push) Has been cancelled
tests / pyright (3.12) (push) Has been cancelled
tests / pyright (3.13) (push) Has been cancelled
tests / ty-check (3.10) (push) Has been cancelled
tests / ty-check (3.11) (push) Has been cancelled
tests / ty-check (3.12) (push) Has been cancelled
tests / ty-check (3.13) (push) Has been cancelled
tests / stubs (push) Has been cancelled
tests / smoke-test (push) Has been cancelled
Docker / check_paths (push) Has been cancelled
docs / build (push) Has been cancelled
Docker / build (push) Has been cancelled
Upstream: https://github.com/michaelgillett/mjlab Upstream-Commit: c19f713c415a699a79d71cd96aa13c3104a05047 Upstream-Branch: main
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
"""Test that all tyro.cli() calls include mjlab.TYRO_FLAGS."""
|
|
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
ROOT = Path(__file__).parent.parent
|
|
SEARCH_DIRS = [ROOT / "src", ROOT / "scripts"]
|
|
|
|
|
|
def find_tyro_cli_calls() -> list[tuple[Path, int, bool]]:
|
|
"""Find all tyro.cli() calls and check if they have TYRO_FLAGS.
|
|
|
|
Returns a list of (filepath, line_number, has_tyro_flags) tuples.
|
|
"""
|
|
results = []
|
|
|
|
for search_dir in SEARCH_DIRS:
|
|
if not search_dir.exists():
|
|
continue
|
|
for filepath in search_dir.rglob("*.py"):
|
|
try:
|
|
source = filepath.read_text()
|
|
tree = ast.parse(source, filename=str(filepath))
|
|
except SyntaxError:
|
|
continue
|
|
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.Call):
|
|
continue
|
|
|
|
# Check if this is a tyro.cli() call
|
|
func = node.func
|
|
is_tyro_cli = False
|
|
|
|
if isinstance(func, ast.Attribute) and func.attr == "cli":
|
|
# tyro.cli(...)
|
|
if isinstance(func.value, ast.Name) and func.value.id == "tyro":
|
|
is_tyro_cli = True
|
|
elif isinstance(func, ast.Name) and func.id == "cli":
|
|
# from tyro import cli; cli(...)
|
|
is_tyro_cli = True
|
|
|
|
if not is_tyro_cli:
|
|
continue
|
|
|
|
# Check if config= keyword argument contains TYRO_FLAGS
|
|
has_tyro_flags = False
|
|
for keyword in node.keywords:
|
|
if keyword.arg == "config":
|
|
config_source = ast.unparse(keyword.value)
|
|
if "TYRO_FLAGS" in config_source:
|
|
has_tyro_flags = True
|
|
break
|
|
|
|
results.append((filepath, node.lineno, has_tyro_flags))
|
|
|
|
return results
|
|
|
|
|
|
def test_tyro_cli_calls_have_tyro_flags():
|
|
"""Ensure all tyro.cli() calls include config=mjlab.TYRO_FLAGS."""
|
|
all_calls = find_tyro_cli_calls()
|
|
assert all_calls, "No tyro.cli() calls found"
|
|
|
|
violations = [
|
|
(filepath, line) for filepath, line, has_flags in all_calls if not has_flags
|
|
]
|
|
|
|
if violations:
|
|
msg = "tyro.cli() calls missing config=mjlab.TYRO_FLAGS:\n"
|
|
for filepath, line in violations:
|
|
rel_path = filepath.relative_to(ROOT)
|
|
msg += f" {rel_path}:{line}\n"
|
|
pytest.fail(msg)
|