tests/test_triage.py

from champ.triage import safety_triage


def test_breathing_red_flag():
    triggered, reply, reason = safety_triage(
        "My child is having trouble breathing and looks very unwell"
    )
    assert triggered is True
    assert reply is not None
    assert reason == "red_flag_symptoms"


def test_infant_fever_triggers():
    triggered, reply, reason = safety_triage(
        "My 2-week-old has a fever of 38.9"
    )
    assert triggered is True
    assert reply is not None
    assert reason == "infant_under_3_months_with_fever"


def test_non_urgent_case():
    triggered, reply, reason = safety_triage(
        "My 6-year-old has a mild cough and runny nose"
    )
    assert triggered is False
    assert reply is None
    assert reason == ""


def test_follow_up_question_not_triggered():
    triggered, reply, reason = safety_triage(
        "What should I watch for tonight?"
    )
    assert triggered is False
    assert reply is None
    assert reason == ""