medilang-tech / tests /test_helpers.py
Dama03's picture
first push of the AI
411a994
raw
history blame contribute delete
570 Bytes
from app.utils.helpers import normalize_gender, clean_diagnosis, emergency_triage
def test_normalize_gender():
assert normalize_gender("M") == "male"
assert normalize_gender("masculin") == "male"
assert normalize_gender("Femme") == "female"
assert normalize_gender("unknown") is None
def test_clean_diagnosis():
assert clean_diagnosis(" Paludisme ") == "paludisme"
assert clean_diagnosis("") is None
def test_emergency_triage():
assert emergency_triage("douleur poitrine") is True
assert emergency_triage("tout va bien") is False