File size: 870 Bytes
a2ffabc | 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 | import json
from arm_gym.errors import ErrorKind, StructuredError, VerifierResult
def test_structured_error_round_trips_to_json_for_prompt():
e = StructuredError(kind=ErrorKind.ASSEMBLE_FAIL, message="bad token",
line=3, column=5, extra={"token": "xzrq"})
payload = json.loads(e.to_prompt())
assert payload["kind"] == "assemble_fail"
assert payload["line"] == 3
assert payload["column"] == 5
assert payload["extra"]["token"] == "xzrq"
def test_empty_fields_omitted():
e = StructuredError(kind=ErrorKind.SEGFAULT, message="boom")
payload = json.loads(e.to_prompt())
assert "line" not in payload
assert "extra" not in payload
def test_verifier_result_defaults_fail_closed():
v = VerifierResult(ok=False, reward=0.0)
assert v.ok is False
assert v.reward == 0.0
assert v.speedup is None
|