Use ctypes

tests/python/test_basic.py

@@ -29,6 +29,26 @@ def test_basic():
     # assert they are the same
     assert np.sum(np.abs(preds2-preds)) == 0
 
+def test_feature_names():
+    data = np.random.randn(100, 5)
+    target = np.array([0, 1] * 50)
+
+    features = ['Feature1', 'Feature2', 'Feature3', 'Feature4', 'Feature5']
+    dm = xgb.DMatrix(data, label=target,
+                     feature_names=features)
+    assert dm.feature_names == features
+    assert dm.num_row() == 100
+    assert dm.num_col() == 5
+
+    params={'objective': 'multi:softprob',
+            'eval_metric': 'mlogloss',
+            'eta': 0.3,
+            'num_class': 3}
+
+    bst = xgb.train(params, dm, num_boost_round=10)
+    scores = bst.get_fscore()
+    assert list(sorted(k for k in scores)) == features
+
 def test_plotting():
     bst2 = xgb.Booster(model_file='xgb.model')
     # plotting