mirror of https://github.com/explosion/spaCy.git
Change test595 to mock data, instead of requiring model.
This commit is contained in:
parent
a4eb5c2bff
commit
0595cc0635
|
@ -1,12 +1,41 @@
|
|||
from __future__ import unicode_literals
|
||||
import pytest
|
||||
|
||||
import spacy
|
||||
from ...symbols import POS, VERB, VerbForm_inf
|
||||
from ...tokens import Doc
|
||||
from ...vocab import Vocab
|
||||
from ...lemmatizer import Lemmatizer
|
||||
|
||||
|
||||
@pytest.mark.models
|
||||
def test_not_lemmatize_base_forms():
|
||||
nlp = spacy.load('en', parser=False)
|
||||
doc = nlp(u"Don't feed the dog")
|
||||
@pytest.fixture
|
||||
def index():
|
||||
return {'verb': {}}
|
||||
|
||||
@pytest.fixture
|
||||
def exceptions():
|
||||
return {'verb': {}}
|
||||
|
||||
@pytest.fixture
|
||||
def rules():
|
||||
return {"verb": [["ed", "e"]]}
|
||||
|
||||
@pytest.fixture
|
||||
def lemmatizer(index, exceptions, rules):
|
||||
return Lemmatizer(index, exceptions, rules)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def tag_map():
|
||||
return {'VB': {POS: VERB, 'morph': VerbForm_inf}}
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def vocab(lemmatizer, tag_map):
|
||||
return Vocab(lemmatizer=lemmatizer, tag_map=tag_map)
|
||||
|
||||
|
||||
def test_not_lemmatize_base_forms(vocab, lemmatizer):
|
||||
doc = Doc(vocab, words=["Do", "n't", "feed", "the", "dog"])
|
||||
feed = doc[2]
|
||||
feed.tag_ = u'VB'
|
||||
assert feed.text == u'feed'
|
||||
|
|
Loading…
Reference in New Issue