spaCy/spacy/tests/lang/fr/test_lemmatization.py

# coding: utf-8
from __future__ import unicode_literals

import pytest


@pytest.mark.models('fr')
def test_lemmatizer_verb(FR):
    tokens = FR("Qu'est-ce que tu fais?")
    assert tokens[0].lemma_ == "que"
    assert tokens[1].lemma_ == "être"
    assert tokens[5].lemma_ == "faire"


@pytest.mark.models('fr')
@pytest.mark.xfail(reason="sont tagged as AUX")
def test_lemmatizer_noun_verb_2(FR):
    tokens = FR("Les abaissements de température sont gênants.")
    assert tokens[4].lemma_ == "être"


@pytest.mark.models('fr')
@pytest.mark.xfail(reason="Costaricienne TAG is PROPN instead of NOUN and spacy don't lemmatize PROPN")
def test_lemmatizer_noun(model):
    tokens = FR("il y a des Costaricienne.")
    assert tokens[4].lemma_ == "Costaricain"


@pytest.mark.models('fr')
def test_lemmatizer_noun_2(FR):
    tokens = FR("Les abaissements de température sont gênants.")
    assert tokens[1].lemma_ == "abaissement"
    assert tokens[5].lemma_ == "gênant"