option to shuffle data in mknfolds (#1459)

* option to shuffle data in mknfolds * removed possibility to run as stand alone test * split function def in 2 lines for lint * option to shuffle data in mknfolds * removed possibility to run as stand alone test * split function def in 2 lines for lint
2016-12-22 17:53:30 -06:00
parent b49b339183
commit fb0fc0c580
2 changed files with 22 additions and 6 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -241,3 +241,12 @@ class TestBasic(unittest.TestCase):
        cv = xgb.cv(params, dm, num_boost_round=10, nfold=10, as_pandas=False)
        assert isinstance(cv, dict)
        assert len(cv) == (4)
+
+    def test_cv_no_shuffle(self):
+        dm = xgb.DMatrix(dpath + 'agaricus.txt.train')
+        params = {'max_depth': 2, 'eta': 1, 'silent': 1, 'objective': 'binary:logistic'}
+
+        # return np.ndarray
+        cv = xgb.cv(params, dm, num_boost_round=10, shuffle=False, nfold=10, as_pandas=False)
+        assert isinstance(cv, dict)
+        assert len(cv) == (4)