import pytest

import edsnlp


def test_pipelines(doc):
    assert len(doc.ents) == 3
    patient, _, anomalie = doc.ents

    assert not patient._.negation
    assert anomalie._.negation

    assert not doc[0]._.history


def test_import_all():
    import edsnlp.pipes

    for name in dir(edsnlp.pipes):
        if not name.startswith("_") and "endlines" not in name:
            try:
                getattr(edsnlp.pipes, name)
            except (ImportError, AttributeError) as e:
                if "torch" in str(e):
                    pass


def test_non_existing_pipe():
    with pytest.raises(AttributeError) as e:
        getattr(edsnlp.pipes, "non_existing_pipe")

    assert str(e.value) == "module edsnlp.pipes has no attribute non_existing_pipe"