Change to properties

tests/python/test_basic.py

@@ -48,21 +48,23 @@ class TestBasic(unittest.TestCase):
                           feature_names=['a', 'b', 'c', 'd', 'e=1'])
 
         dm = xgb.DMatrix(data)
-        dm.set_feature_names(list('abcde'))
-        assert dm.get_feature_names() == list('abcde')
+        dm.feature_names = list('abcde')
+        assert dm.feature_names == list('abcde')
 
-        dm.set_feature_types('q')
-        assert dm.get_feature_types() == list('qqqqq')
+        dm.feature_types = 'q'
+        assert dm.feature_types == list('qqqqq')
 
-        dm.set_feature_types(list('qiqiq'))
-        assert dm.get_feature_types() == list('qiqiq')
+        dm.feature_types = list('qiqiq')
+        assert dm.feature_types == list('qiqiq')
 
-        self.assertRaises(ValueError, dm.set_feature_types, list('abcde'))
+        def incorrect_type_set():
+            dm.feature_types = list('abcde')
+        self.assertRaises(ValueError, incorrect_type_set)
 
         # reset
-        dm.set_feature_names(None)
-        assert dm.get_feature_names() is None
-        assert dm.get_feature_types() is None
+        dm.feature_names = None
+        assert dm.feature_names is None
+        assert dm.feature_types is None
 
     def test_feature_names(self):
         data = np.random.randn(100, 5)