explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-25 00:34:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix test_basic_create

Browse Source
This commit is contained in:
Matthew Honnibal 2015-11-07 10:03:56 +11:00
parent cc8febcbe1
commit 1d3884c46d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/tests/test_basic_create.py
View File

@ -11,8 +11,8 @@ from spacy.strings import StringStore
from spacy.vocab import Vocab from spacy.vocab import Vocab
from spacy.tokenizer import Tokenizer from spacy.tokenizer import Tokenizer
from spacy.syntax.arc_eager import ArcEager from spacy.syntax.arc_eager import ArcEager
from spacy.tagger import Tagger from spacy.tagger import Tagger, TaggerModel
from spacy.syntax.parser import Parser from spacy.syntax.parser import Parser, ParserModel
from spacy.matcher import Matcher from spacy.matcher import Matcher
@ -63,7 +63,7 @@ class TestTagger(unittest.TestCase):
def test_create(self): def test_create(self):
vocab = Vocab() vocab = Vocab()
templates = ((1,),) templates = ((1,),)
model = Model(vocab.morphology.n_tags, templates, model_loc=None) model = TaggerModel(vocab.morphology.n_tags, templates, model_loc=None)
tagger = Tagger(vocab, model) tagger = Tagger(vocab, model)
@ -73,7 +73,7 @@ class TestParser(unittest.TestCase):
templates = ((1,),) templates = ((1,),)
labels_by_action = {0: ['One', 'Two'], 1: ['Two', 'Three']} labels_by_action = {0: ['One', 'Two'], 1: ['Two', 'Three']}
transition_system = ArcEager(vocab.strings, labels_by_action) transition_system = ArcEager(vocab.strings, labels_by_action)
model = Model(vocab.morphology.n_tags, templates, model_loc=None) model = ParserModel(transition_system.n_moves, templates)
parser = Parser(vocab.strings, transition_system, model) parser = Parser(vocab.strings, transition_system, model)