fix LdaMallet test

gojomo · Apr 10, 2015 · d97da08 · d97da08
1 parent 05aee93
commit d97da08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gensim/test/test_models.py b/gensim/test/test_models.py
@@ -513,7 +513,7 @@ def testLargeMmapCompressed(self):
         model.save(fname, sep_limit=0)
 
         # test loading the large model arrays with mmap
-        self.assertRaises(IOError, ldamodel.LdaModel.load, fname, mmap=None)
+        self.assertRaises(IOError, ldamodel.LdaModel.load, fname, mmap='r')
 #endclass TestLdaMallet
 
 
@@ -529,7 +529,7 @@ def testTransform(self):
         doc = list(self.corpus)[0]
         transformed = model[doc]
 
-        expected =  [(0, 0.57735026918962573), (1, 0.57735026918962573), (2, 0.57735026918962573)]
+        expected = [(0, 0.57735026918962573), (1, 0.57735026918962573), (2, 0.57735026918962573)]
         self.assertTrue(numpy.allclose(transformed, expected))