polyguard-openenv-workbench / polyguard-rl /tests /test_dataops_parser.py
TheJackBright's picture
Deploy GitHub root master to Space
c296d62
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")