Add python test.

dmlc · Feb 16, 2019 · 4924640 · 4924640
1 parent 509b683
commit 4924640
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/python/test_basic.py b/tests/python/test_basic.py
@@ -63,6 +63,12 @@ def test_basic(self):
         # assert they are the same
         assert np.sum(np.abs(preds2 - preds)) == 0
 
+    def test_np_view(self):
+        y = np.array([12, 34, 56], np.float32)[::2]
+        from_view = xgb.DMatrix([], label=y).get_label()
+        from_array = xgb.DMatrix([], label=y+0).get_label()
+        assert (from_view == from_array).all()
+
     def test_record_results(self):
         dtrain = xgb.DMatrix(dpath + 'agaricus.txt.train')
         dtest = xgb.DMatrix(dpath + 'agaricus.txt.test')