adithya9903's picture
Deploy PolyGuard HF training Space
fd0c71a verified
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")