set cleanup_files True for wordrank tests

piskvorky · Jun 12, 2017 · 2a5d048 · 2a5d048
1 parent b5210ab
commit 2a5d048
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"""