mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-25 17:36:30 +03:00
Add test for #1727
This commit is contained in:
parent
85c942a6e3
commit
fd187d71ad
23
spacy/tests/regression/test_issue1727.py
Normal file
23
spacy/tests/regression/test_issue1727.py
Normal file
|
@ -0,0 +1,23 @@
|
|||
from __future__ import unicode_literals
|
||||
import numpy
|
||||
from ...pipeline import Tagger
|
||||
from ...vectors import Vectors
|
||||
from ...vocab import Vocab
|
||||
from ..util import make_tempdir
|
||||
|
||||
|
||||
def test_issue1727():
|
||||
data = numpy.ones((3, 300), dtype='f')
|
||||
keys = [u'I', u'am', u'Matt']
|
||||
vectors = Vectors(data=data, keys=keys)
|
||||
tagger = Tagger(Vocab())
|
||||
tagger.add_label('PRP')
|
||||
tagger.begin_training()
|
||||
|
||||
assert tagger.cfg.get('pretrained_dims', 0) == 0
|
||||
tagger.vocab.vectors = vectors
|
||||
|
||||
with make_tempdir() as path:
|
||||
tagger.to_disk(path)
|
||||
tagger = Tagger(Vocab()).from_disk(path)
|
||||
assert tagger.cfg.get('pretrained_dims', 0) == 0
|
Loading…
Reference in New Issue
Block a user