Add check for length of weights. (#4872)

2019-12-21 11:30:58 +08:00
parent 3d04a8cc97
commit c8bdb652c4
4 changed files with 41 additions and 3 deletions
--- a/tests/cpp/objective/test_multiclass_obj.cc
+++ b/tests/cpp/objective/test_multiclass_obj.cc
@@ -25,6 +25,13 @@ TEST(Objective, DeclareUnifiedTest(SoftmaxMultiClassObjGPair)) {
 		   {0.24f, -0.91f, 0.66f, -0.33f, 0.09f, 0.24f}, // grad
 		   {0.36f, 0.16f, 0.44f, 0.45f, 0.16f, 0.37f});	 // hess

+  CheckObjFunction(obj,
+		   {1.0f, 0.0f, 2.0f, 2.0f, 0.0f, 1.0f}, // preds
+		   {1.0f, 0.0f},	       // labels
+                   {},                         // weights
+		   {0.24f, -0.91f, 0.66f, -0.33f, 0.09f, 0.24f}, // grad
+		   {0.36f, 0.16f, 0.44f, 0.45f, 0.16f, 0.37f});	 // hess
+
  ASSERT_NO_THROW(obj->DefaultEvalMetric());
 }