Fix #3714: preserve feature names when slicing DMatrix (#3766)

* Fix #3714: preserve feature names when slicing DMatrix * Add test
2018-10-08 01:04:33 -07:00
parent 813d2436d3
commit 10cd7c8447
2 changed files with 9 additions and 1 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -118,6 +118,8 @@ class TestBasic(unittest.TestCase):
        dm.feature_names = list('abcde')
        assert dm.feature_names == list('abcde')

+        assert dm.slice([0, 1]).feature_names == dm.feature_names
+
        dm.feature_types = 'q'
        assert dm.feature_types == list('qqqqq')