Use only one thread in test_basic.py (microsoft#412)

Alnusjaponica · Apr 13, 2017 · a5f11d4 · a5f11d4
1 parent ab55910
commit a5f11d4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python_package_test/test_basic.py b/tests/python_package_test/test_basic.py
@@ -22,7 +22,8 @@ def test(self):
             "metric": "auc",
             "min_data": 10,
             "num_leaves": 15,
-            "verbose": -1
+            "verbose": -1,
+            "num_threads": 1
         }
         bst = lgb.Booster(params, train_data)
         bst.add_valid(valid_data, "valid_1")
@@ -47,6 +48,7 @@ def test(self):
         pred_from_model_file = bst.predict(X_test)
         self.assertEqual(len(pred_from_matr), len(pred_from_model_file))
         for preds in zip(pred_from_matr, pred_from_model_file):
+            # we need to check the consistency of model file here, so test for exact equal
             self.assertEqual(*preds)
         # check pmml
         os.system('python ../../pmml/pmml.py model.txt')