1#@!#!123s

: / usr / local / lib / python3.6 / site-packages / spacy_lookups_data / tests /

Filename : test_fr.py

back

# coding: utf-8
from __future__ import unicode_literals

import pytest
from spacy.lang.fr import French


@pytest.fixture(scope="session")
def fr_nlp():
    return French()


def test_fr_lemmatizer_verb(fr_nlp):
    tokens = fr_nlp("Qu'est-ce que tu fais?")
    assert tokens[0].lemma_ == "que"
    assert tokens[1].lemma_ == "être"
    assert tokens[5].lemma_ == "faire"


def test_fr_lemmatizer_noun_verb_2(fr_nlp):
    tokens = fr_nlp("Les abaissements de température sont gênants.")
    assert tokens[4].lemma_ == "être"


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


def test_fr_lemmatizer_noun_2(fr_nlp):
    tokens = fr_nlp("Les abaissements de température sont gênants.")
    assert tokens[1].lemma_ == "abaissement"
    assert tokens[5].lemma_ == "gênant"