test_word2vec.py 文件源码

def test_serializzation(small_model):
    with io.BytesIO() as fileobj:
        pickle.dump(small_model, fileobj)
        fileobj.seek(0)
        loaded = pickle.load(fileobj)
        assert all(str(loaded.wv.vocab[w]) == str(small_model.wv.vocab[w]) for w in small_model.wv.vocab)
        assert all(str(loaded.lvocab[w]) == str(small_model.lvocab[w]) for w in small_model.lvocab)
        assert numpy.array_equiv(loaded.syn1, small_model.syn1)
        assert numpy.array_equiv(loaded.wv.syn0, small_model.wv.syn0)