Bug mixing DMatrix's with and without feature names

2016-04-29 13:51:34 +09:00
parent ff4dda2102
commit 6bab164d80
4 changed files with 25 additions and 14 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -91,7 +91,7 @@ class TestBasic(unittest.TestCase):

        # reset
        dm.feature_names = None
-        assert dm.feature_names is None
+        self.assertEqual(dm.feature_names, ['f0', 'f1', 'f2', 'f3', 'f4'])
        assert dm.feature_types is None

    def test_feature_names(self):