mirror of https://github.com/explosion/spaCy.git
18 lines
472 B
Python
18 lines
472 B
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.models
|
|
@pytest.mark.parametrize('text1,text2',
|
|
[("You're happy", "You are happy"),
|
|
("I'm happy", "I am happy"),
|
|
("he's happy", "he's happy")])
|
|
def test_issue717(EN, text1, text2):
|
|
"""Test that contractions are assigned the correct lemma."""
|
|
doc1 = EN(text1)
|
|
doc2 = EN(text2)
|
|
assert doc1[1].lemma_ == doc2[1].lemma_
|
|
assert doc1[1].lemma == doc2[1].lemma
|