polyguard-rl/tests/test_dataops_parser.py

from app.dataops.parser import extract_components, extract_drug_mentions


def test_extract_components_accepts_hyphenated_tokens() -> None:
    text = "Active ingredients: beta-blocker, ace inhibitor; other text"

    assert "beta-blocker" in extract_components(text)


def test_extract_drug_mentions_accepts_hyphenated_tokens() -> None:
    assert "beta-blocker" in extract_drug_mentions("beta-blocker")