from edsnlp.core import PipelineProtocol
from edsnlp.utils.examples import parse_example

examples = [
    "Patient admis pour <ent code=A01>fièvres typhoïde et paratyphoïde</ent>",
    "Patient admis pour <ent code=C221>C2.21</ent>",
]


def test_cim10(blank_nlp: PipelineProtocol):
    blank_nlp.add_pipe("eds.cim10")

    for text, entities in map(parse_example, examples):
        doc = blank_nlp(text)

        assert len(doc.ents) == len(entities)

        for ent, entity in zip(doc.ents, entities):
            assert ent.text == text[entity.start_char : entity.end_char]
            assert ent.kb_id_ == entity.modifiers[0].value