mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-26 18:06:29 +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