Have a hanging test.

dmlc · Oct 29, 2020 · a15a244 · a15a244
1 parent d0fd4f3
commit a15a244
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/hang b/hang
@@ -0,0 +1 @@
+81234168284343979873719512816958541056
diff --git a/tests/python/test_predict.py b/tests/python/test_predict.py
@@ -38,6 +38,10 @@ def run_predict_leaf(predictor):
          'predictor': predictor, 'tree_method': 'hist'}, m,
         num_boost_round=num_boost_round)
 
+    empty = xgb.DMatrix(np.ones(shape=(0, cols)))
+    empty_leaf = booster.predict(empty, pred_leaf=True)
+    assert empty_leaf.shape[0] == 0
+
     leaf = booster.predict(m, pred_leaf=True)
     assert leaf.shape[0] == rows
     assert leaf.shape[1] == classes * num_parallel_tree * num_boost_round