File size: 1,546 Bytes
a15535e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | """Smoke tests for warm-start pair generation."""
import json
import tempfile
from pathlib import Path
from warmstart.generate_pairs import generate_pairs
def test_generate_pairs_produces_minimum_count():
with tempfile.TemporaryDirectory() as tmp:
counts = generate_pairs(target_count=50, out_dir=Path(tmp))
assert counts["repair_pairs"] >= 50
assert counts["drift_pairs"] >= 50
repair_jsonl = Path(tmp) / "repair_pairs.jsonl"
drift_jsonl = Path(tmp) / "drift_pairs.jsonl"
assert repair_jsonl.exists()
assert drift_jsonl.exists()
first = json.loads(repair_jsonl.read_text(encoding="utf-8").splitlines()[0])
assert first["role_target"] == "repair_agent"
assert "messages" in first and len(first["messages"]) == 3
assert first["messages"][-1]["content"] # non-empty assistant content
first_drift = json.loads(drift_jsonl.read_text(encoding="utf-8").splitlines()[0])
assert first_drift["role_target"] == "drift_generator"
body = first_drift["messages"][-1]["content"]
parsed = json.loads(body)
assert "primitive_type" in parsed and "params" in parsed
def test_generate_pairs_covers_multiple_primitive_types():
with tempfile.TemporaryDirectory() as tmp:
generate_pairs(target_count=50, out_dir=Path(tmp))
summary = json.loads((Path(tmp) / "summary.json").read_text(encoding="utf-8"))
assert len(summary["primitives_covered"]) >= 5
assert len(summary["tasks_covered"]) >= 5
|