Merge pull request #1410 from parulsethi/fix_wordrank_test

Fix wordrank tests
piskvorky · Jun 13, 2017 · 85d8ba1 · 85d8ba1
2 parents 728697a + 2a5d048
commit 85d8ba1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/test/test_wordrank_wrapper.py b/gensim/test/test_wordrank_wrapper.py
@@ -34,7 +34,7 @@ def setUp(self):
         self.wr_file = datapath('test_glove.txt')
         if not self.wr_path:
             return
-        self.test_model = wordrank.Wordrank.train(self.wr_path, self.corpus_file, self.out_name, iter=6, dump_period=5, period=5)
+        self.test_model = wordrank.Wordrank.train(self.wr_path, self.corpus_file, self.out_name, iter=6, dump_period=5, period=5, cleanup_files=True)
 
     def testLoadWordrankFormat(self):
         """Test model successfully loaded from Wordrank format file"""