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

* Add tests for pickling vectors, re: Issue #125

Browse Source
This commit is contained in:
Matthew Honnibal 2015-10-26 12:31:05 +11:00
parent 314090cc78
commit a824a98312
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
spacy/tests/vocab/test_vocab.py
View File

@ -52,3 +52,33 @@ def test_pickle_vocab(en_vocab):
file_.seek(0) file_.seek(0)
loaded = pickle.load(file_) loaded = pickle.load(file_)
@pytest.mark.vectors
def test_pickle_vocab_vectors(en_vocab):
vectors_length = en_vocab.vectors_length
assert vectors_length != 0
apples = en_vocab['apples']
oranges = en_vocab['oranges']
hippos = en_vocab['hippos']
assert apples.similarity(oranges) > apples.similarity(hippos)
apples.vector = hippos.vector
assert apples.similarity(oranges) < apples.similarity(hippos)
file_ = io.BytesIO()
cloudpickle.dump(en_vocab, file_)
file_.seek(0)
loaded = pickle.load(file_)
apples = loaded['apples']
oranges = loaded['oranges']
hippos = loaded['hippos']
assert apples.similarity(oranges) < apples.similarity(hippos)