spaCy/spacy/tests/regression/test_issue781.py
2017-11-06 17:34:50 +01:00

13 lines
478 B
Python

# coding: utf-8
from __future__ import unicode_literals
import pytest
# Note: "chromosomes" worked previous the bug fix
@pytest.mark.models('en')
@pytest.mark.parametrize('word,lemmas', [("chromosomes", ["chromosome"]), ("endosomes", ["endosome"]), ("colocalizes", ["colocaliz", "colocalize"])])
def test_issue781(EN, word, lemmas):
lemmatizer = EN.Defaults.create_lemmatizer()
assert sorted(lemmatizer(word, 'noun', morphology={'number': 'plur'})) == sorted(lemmas)