Fix get_uint_info() (#3442)

* Add regression test
2018-07-05 20:06:59 -07:00
parent 48d6e68690
commit 66e74d2223
2 changed files with 17 additions and 3 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -299,3 +299,10 @@ class TestBasic(unittest.TestCase):
            )
            output = out.getvalue().strip()
        assert output == '[array([5., 8.], dtype=float32), array([23., 43., 11.], dtype=float32)]'
+
+    def test_get_info(self):
+        dtrain = xgb.DMatrix(dpath + 'agaricus.txt.train')
+        dtrain.get_float_info('label')
+        dtrain.get_float_info('weight')
+        dtrain.get_float_info('base_margin')
+        dtrain.get_uint_info('root_index')