def test__calc_flex_raises_error_if_non_valid_value(
nlp: Language, searcher: FuzzySearcher
) -> None:
"""It raises TypeError if flex is not an int or "default"."""
query = nlp("Test query.")
with pytest.raises(TypeError):
searcher._calc_flex(query, None)
def test__calc_flex_warns_if_flex_longer_than_query(
nlp: Language, searcher: FuzzySearcher
) -> None:
"""It provides UserWarning if flex > len(query)."""
query = nlp.make_doc("Test query.")
with pytest.warns(FlexWarning):
searcher._calc_flex(query, 5)
def test__calc_flex_passes_through_valid_value(
nlp: Language, searcher: FuzzySearcher) -> None:
"""It passes through a valid flex value (<= len(query))."""
query = nlp.make_doc("Test query.")
assert searcher._calc_flex(query, 1) == 1
def test__calc_flex_with_default(nlp: Language,
searcher: FuzzySearcher) -> None:
"""It returns len(query) if set with "default"."""
query = nlp.make_doc("Test query.")
assert searcher._calc_flex(query, "default") == 3
