explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-13 05:07:03 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix pickle tests

Browse Source
This commit is contained in:
Matthew Honnibal 2018-12-06 20:46:36 +01:00
parent e619f45287
commit bb3304a4f1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/tests/test_pickles.py
View File

@ -29,8 +29,8 @@ def test_pickle_vocab(text1, text2):
lex2 = vocab[text2] lex2 = vocab[text2]
assert lex1.norm_ == text1[:-1] assert lex1.norm_ == text1[:-1]
assert lex2.norm_ == text2[:-1] assert lex2.norm_ == text2[:-1]
data = pickle.dumps(vocab) data = srsly.pickle_dumps(vocab)
unpickled = pickle.loads(data) unpickled = srsly.pickle_loads(data)
assert unpickled[text1].orth == lex1.orth assert unpickled[text1].orth == lex1.orth
assert unpickled[text2].orth == lex2.orth assert unpickled[text2].orth == lex2.orth
assert unpickled[text1].norm == lex1.norm assert unpickled[text1].norm == lex1.norm