minor fix

piskvorky · Oct 27, 2018 · 368d264 · 368d264
1 parent c17ebd0
commit 368d264
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gensim/test/test_corpora.py b/gensim/test/test_corpora.py
@@ -391,7 +391,8 @@ def test_serialization(self):
         self.assertEqual(serialized_corpus[1], second_corpus)
         self.corpus_class.serialize(path, common_corpus, labels=np.array(labels))
         serialized_corpus = self.corpus_class(path)
-        self.assertEqual(serialized_corpus[1], np.array(second_corpus))
+        second_corpus = np.array(second_corpus)
+        self.assertEqual(serialized_corpus[1], second_corpus)
 
 
 class TestBleiCorpus(CorpusTestCase):