Spaces:
Running
Running
File size: 2,568 Bytes
877add7 | 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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 | import pytest
from app.common.exceptions import ParserError
from app.models.policy.parser import parse_decision, parse_decision_soft, parse_decision_strict_xml, parse_decision_with_repair
def test_parser_valid_json() -> None:
payload = """
{"mode":"REGIMEN_OPT","action_type":"KEEP_REGIMEN","target_drug":null,"replacement_drug":null,
"dose_bucket":"NA","taper_days":null,"monitoring_plan":null,"candidate_id":"cand_01","confidence":0.7}
"""
parsed = parse_decision(payload)
assert parsed.candidate_id == "cand_01"
def test_parser_invalid_json() -> None:
with pytest.raises(ParserError):
parse_decision("not-json")
def test_parser_xml_format() -> None:
payload = """
<decision>
<mode>REGIMEN_OPT</mode>
<action_type>KEEP_REGIMEN</action_type>
<target_drug>null</target_drug>
<replacement_drug>null</replacement_drug>
<dose_bucket>NA</dose_bucket>
<taper_days>null</taper_days>
<monitoring_plan>null</monitoring_plan>
<candidate_id>cand_01</candidate_id>
<confidence>0.7</confidence>
</decision>
"""
parsed = parse_decision(payload)
assert parsed.candidate_id == "cand_01"
def test_parser_strict_xml_rejects_unknown_field() -> None:
payload = """
<decision>
<mode>REGIMEN_OPT</mode>
<action_type>KEEP_REGIMEN</action_type>
<target_drug>null</target_drug>
<replacement_drug>null</replacement_drug>
<dose_bucket>NA</dose_bucket>
<taper_days>null</taper_days>
<candidate_id>cand_01</candidate_id>
<confidence>0.7</confidence>
<unknown>bad</unknown>
</decision>
"""
with pytest.raises(ParserError):
parse_decision_strict_xml(payload)
def test_parser_soft_recovery_from_buffered_xml() -> None:
payload = "prefix garbage <decision><mode>REGIMEN_OPT</mode><action_type>KEEP_REGIMEN</action_type><target_drug>null</target_drug><replacement_drug>null</replacement_drug><dose_bucket>NA</dose_bucket><taper_days>null</taper_days><candidate_id>cand_01</candidate_id><confidence>0.7</confidence></decision> suffix"
parsed = parse_decision_soft(payload)
assert parsed.action_type.value == "KEEP_REGIMEN"
def test_parser_repair_invalid_candidate_and_confidence() -> None:
payload = "{\"mode\":\"REGIMEN_OPT\",\"action_type\":\"KEEP_REGIMEN\",\"dose_bucket\":\"NA\",\"candidate_id\":\"bad\",\"confidence\":2.4}"
parsed = parse_decision_with_repair(payload)
assert parsed.candidate_id.startswith("cand_")
assert 0.001 <= parsed.confidence <= 0.999
|