def test_multi_match_will_not_match_if_regex_starts_ends_with_space(
searcher: RegexSearcher, nlp: Language) -> None:
"""Regexes that match on spaces will not produce match."""
doc = nlp(
"We want to identify US but will fail because regex includes whitespaces."
)
matches = searcher.match(doc, "\\s[Uu](nited|\\.?) ?[Ss](tates|\\.?)\\s")
assert matches == []
def test_multi_match_will_expand_on_partial_match_if_partials(
searcher: RegexSearcher, nlp: Language) -> None:
"""It extends partial matches to span boundaries."""
doc = nlp(
"We want to identify 'USA' even though only first two letters will matched."
)
matches = searcher.match(doc, "[Uu](nited|\\.?) ?[Ss](tates|\\.?)")
assert matches == [(5, 6, (0, 0, 0))]
def test_multi_match_will_not_expand_if_not_partials(searcher: RegexSearcher,
nlp: Language) -> None:
"""It will not extend partial matches to span boundaries if not partial."""
doc = nlp(
"We want to identify 'USA' even though only first two letters will matched."
)
matches = searcher.match(doc,
"[Uu](nited|\\.?) ?[Ss](tates|\\.?)",
partial=False)
assert matches == []
def test_multi_match_raises_error_if_regex_str_not_str(searcher: RegexSearcher,
nlp: Language) -> None:
"""It raises a type error if regex_str is not a string."""
doc = nlp("My phone number is (555) 555-5555.")
with pytest.raises(TypeError):
searcher.match(doc, 1, predef=True) # type: ignore
def test_multi_match(searcher: RegexSearcher, nlp: Language) -> None:
"""It produces matches."""
doc = nlp("My phone number is (555) 555-5555, not (554) 554-5554.")
matches = searcher.match(doc, "phones", predef=True)
assert matches == [(4, 10, (0, 0, 0)), (12, 18, (0, 0, 0))]
def test_regex_search_has_empty_config_if_empty_passed(nlp: Language) -> None:
"""Its config is empty."""
searcher = RegexSearcher(vocab=nlp.vocab, config="empty")
assert searcher._config._predefs == {}
def test_regexsearcher_raises_error_if_config_is_not_regexconfig(
nlp: Language) -> None:
"""It raises a TypeError if config is not recognized string or RegexConfig."""
with pytest.raises(TypeError):
RegexSearcher(vocab=nlp.vocab, config="Will cause error")
def test_regexsearcher_uses_passed_config(nlp: Language) -> None:
"""It uses the config passed to it."""
config = RegexConfig()
config._predefs["test"] = regex.compile("test")
searcher = RegexSearcher(vocab=nlp.vocab, config=config)
assert "test" in searcher._config._predefs
def searcher(nlp: Language) -> RegexSearcher:
"""It returns a default regex searcher."""
return RegexSearcher(vocab=nlp.vocab)