47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from lai_automation.agent import _SYSTEM_TEMPLATE, MiniSWEAgentRunner
|
|
from lai_automation.models import Issue
|
|
|
|
|
|
def test_task_uses_issue_text_not_github_parser() -> None:
|
|
issue = Issue(
|
|
number=42,
|
|
title="Wrong result",
|
|
body="Original prompt and expected result",
|
|
html_url="https://gitea.example.test/owner/lai/issues/42",
|
|
labels=(),
|
|
)
|
|
|
|
task = MiniSWEAgentRunner._task(issue)
|
|
|
|
assert "Issue #42: Wrong result" in task
|
|
assert issue.html_url in task
|
|
assert issue.body in task
|
|
|
|
|
|
def test_task_includes_validation_feedback_for_repair_attempt() -> None:
|
|
issue = Issue(
|
|
number=43,
|
|
title="Broken query",
|
|
body="Original report",
|
|
html_url="https://gitea.example.test/owner/lai/issues/43",
|
|
labels=(),
|
|
)
|
|
|
|
task = MiniSWEAgentRunner._task(issue, "PHP parse error on line 3")
|
|
|
|
assert "previous attempt failed deterministic validation" in task
|
|
assert "PHP parse error on line 3" in task
|
|
|
|
|
|
def test_system_prompt_requires_precise_diff_hygiene() -> None:
|
|
assert "Never create backup files" in _SYSTEM_TEMPLATE
|
|
assert "failure-report transport code" in _SYSTEM_TEMPLATE
|
|
assert "Preserve unrelated lines exactly" in _SYSTEM_TEMPLATE
|
|
assert "small localized edits around a uniquely matching existing anchor" in _SYSTEM_TEMPLATE
|
|
assert "Never reconstruct or rewrite an entire structured file" in _SYSTEM_TEMPLATE
|
|
assert "closest existing test class" in _SYSTEM_TEMPLATE
|
|
assert "inspect all key columns and cardinality" in _SYSTEM_TEMPLATE
|
|
assert "remove trailing spaces" in _SYSTEM_TEMPLATE
|