80e9e26deca028ef6b1b51c307b61bb305dc9b1c,txtgen/data/embedding_test.py,EmbeddingTest,test_load_glove,#EmbeddingTest#,23

Before Change

        word_vecs, vector_size = embedding.load_glove(glove_file.name)

        self.assertEqual(len(word_vecs), 2)
        self.assertEqual(set(word_vecs.keys()),
                         set(["word", "词".encode("utf-8")]))
        self.assertEqual(vector_size, 3)
        np.testing.assert_array_equal(word_vecs["word"], [1.2, 3.4, 5.6])

    def test_load_word2vec(self):

After Change

        glove_file.write("\n".join(word_vec_lines).encode("utf-8"))
        glove_file.flush()
        vocab = {"word": 0, "词".encode("utf8"): 1}
        word_vecs = np.zeros([2, 3])

        word_vecs = embedding.load_glove(glove_file.name, vocab, word_vecs)

        self.assertEqual(word_vecs.shape[0], 2)

Instances