ai-content-maker/.venv/Lib/site-packages/spacy/tests/lang/ko/test_lemmatization.py

10 lines
309 B
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
import pytest
@pytest.mark.parametrize(
"word,lemma", [("새로운", "새롭"), ("빨간", "빨갛"), ("클수록", ""), ("뭡니까", ""), ("됐다", "")]
)
def test_ko_lemmatizer_assigns(ko_tokenizer, word, lemma):
test_lemma = ko_tokenizer(word)[0].lemma_
assert test_lemma == lemma