diff --git a/spacy/tests/regression/test_issue781.py b/spacy/tests/regression/test_issue781.py index 56a1a9ef0..8a98b48b8 100644 --- a/spacy/tests/regression/test_issue781.py +++ b/spacy/tests/regression/test_issue781.py @@ -5,6 +5,6 @@ import pytest # Note: "chromosomes" worked previous the bug fix -@pytest.mark.parametrize('word,lemma', [("chromosomes", "chromosome"), ("endosomes", "endosome"), ("colocalizes", "colocalize")]) -def test_issue781(path, lemmatizer, word, lemma): - assert lemmatizer(word, 'noun', morphology={'number': 'plur'}) == set([lemma]) +@pytest.mark.parametrize('word,lemmas', [("chromosomes", ["chromosome"]), ("endosomes", ["endosome"]), ("colocalizes", ["colocalize", "colocaliz"])]) +def test_issue781(path, lemmatizer, word, lemmas): + assert lemmatizer(word, 'noun', morphology={'number': 'plur'}) == set(lemmas)