a6dc10dad2* Reduce and tighten comments and docstrings in tests Shorten verbose comments and docstrings across the test suite without changing any test logic. Remove narration that restates the next line, collapse long module and test docstrings to a single line, and drop banner separators. Keep regression context (issue and PR references, run ids), skip reasons, mocking and timing rationale, license headers, lint and type directives, and commented-out code. Comments and docstrings only: an AST signature check confirms no code, assertions, or string literals changed, and the suite byte-compiles cleanly. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""AST-based tests that `unsloth run` is registered as a top-level alias for `unsloth studio run`."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
_CLI_INIT = Path(__file__).resolve().parents[2] / "unsloth_cli" / "__init__.py"
|
|
|
|
|
|
def _module_calls(source: str):
|
|
tree = ast.parse(source)
|
|
for node in ast.walk(tree):
|
|
if isinstance(node, ast.Call):
|
|
yield node
|
|
|
|
|
|
def test_top_level_run_alias_registered():
|
|
"""`app.command("run", ...)` must be invoked with studio_run as its target."""
|
|
source = _CLI_INIT.read_text()
|
|
|
|
# Find ``app.command("run", ...)`` call -- the decorator-call form.
|
|
found_decorator_call = False
|
|
for call in _module_calls(source):
|
|
# Match ``app.command(...)`` syntactically.
|
|
if not (
|
|
isinstance(call.func, ast.Attribute)
|
|
and call.func.attr == "command"
|
|
and isinstance(call.func.value, ast.Name)
|
|
and call.func.value.id == "app"
|
|
):
|
|
continue
|
|
# "run" appears as the first positional arg or as keyword name="run".
|
|
first_pos = call.args[0] if call.args else None
|
|
keyword_name = next((kw.value for kw in call.keywords if kw.arg == "name"), None)
|
|
is_run = (isinstance(first_pos, ast.Constant) and first_pos.value == "run") or (
|
|
isinstance(keyword_name, ast.Constant) and keyword_name.value == "run"
|
|
)
|
|
if is_run:
|
|
found_decorator_call = True
|
|
break
|
|
assert (
|
|
found_decorator_call
|
|
), 'Expected `app.command("run", ...)` registration in unsloth_cli/__init__.py'
|
|
|
|
|
|
def test_studio_run_imported_for_alias():
|
|
"""The alias must wire up to the studio.run function, not redefine it."""
|
|
source = _CLI_INIT.read_text()
|
|
tree = ast.parse(source)
|
|
has_import = False
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.ImportFrom):
|
|
continue
|
|
if node.module != "unsloth_cli.commands.studio":
|
|
continue
|
|
for alias in node.names:
|
|
if alias.name == "run":
|
|
has_import = True
|
|
break
|
|
assert (
|
|
has_import
|
|
), "Expected `from unsloth_cli.commands.studio import run` in unsloth_cli/__init__.py"
|