def test_window_size(spacy_doc):
result_2 = kt.yake(spacy_doc, window_size=2)
result_4 = kt.yake(spacy_doc, window_size=4)
assert len(result_2) > 0 and len(result_4) > 0
assert result_2 != result_4
def test_empty_doc(empty_spacy_doc):
result = kt.yake(empty_spacy_doc)
assert isinstance(result, list)
assert len(result) == 0
def test_n_topn(spacy_doc):
for n in (5, 25):
result = kt.yake(spacy_doc, topn=n)
assert 0 < len(result) <= n
def test_topn_float(spacy_doc):
result = kt.yake(spacy_doc, topn=0.2)
assert len(result) > 0
with pytest.raises(ValueError):
_ = kt.yake(spacy_doc, topn=2.0)
def test_ngrams_2_3(spacy_doc):
result = kt.yake(spacy_doc, ngrams=(2, 3))
assert len(result) > 0
assert all(2 <= len(term.split()) <= 3 for term, _ in result)
def test_ngrams_1(spacy_doc):
result = kt.yake(spacy_doc, ngrams=1)
assert len(result) > 0
assert all(len(term.split()) == 1 for term, _ in result)
def test_normalize_lemma(spacy_doc):
result = kt.yake(spacy_doc, normalize="lemma")
assert len(result) > 0
assert any(term != term.lower() for term, _ in result)
def test_normalize_lower(spacy_doc):
result = kt.yake(spacy_doc, normalize="lower")
assert len(result) > 0
assert all(term == term.lower() for term, _ in result)
def test_default(spacy_doc):
result = kt.yake(spacy_doc)
assert isinstance(result, list) and len(result) > 0
assert all(isinstance(ts, tuple) and len(ts) == 2 for ts in result)
assert all(
isinstance(ts[0], str) and isinstance(ts[1], float) for ts in result)