Fixed unit test

andrewherren · andrewherren · commit 27bf2dd94d7c · 2024-05-06T23:15:49.000-05:00
diff --git a/test/test_tree.cpp b/test/test_tree.cpp
@@ -172,7 +172,7 @@ TEST(Tree, SparseLeafRepresentation) {
 
   // Predict leaf indices of each observation in `dataset`
   std::vector<int32_t> leaf_index_preds(n);
-  tree.PredictLeafIndexInplace(&dataset, leaf_index_preds, 0);
+  tree.PredictLeafIndexInplace(&dataset, leaf_index_preds, 0, 0);
   std::vector<int32_t> leaf_index_expected{1,1,0,1,1,1,1,1,0,0};
   ASSERT_EQ(leaf_index_expected, leaf_index_preds);
 }

Original file line number	Diff line number	Diff line change
`@@ -172,7 +172,7 @@ TEST(Tree, SparseLeafRepresentation) {`
`172`	`172`
`173`	`173`	// Predict leaf indices of each observation in `dataset`
`174`	`174`	`std::vector<int32_t> leaf_index_preds(n);`
`175`		`- tree.PredictLeafIndexInplace(&dataset, leaf_index_preds, 0);`
	`175`	`+ tree.PredictLeafIndexInplace(&dataset, leaf_index_preds, 0, 0);`
`176`	`176`	`std::vector<int32_t> leaf_index_expected{1,1,0,1,1,1,1,1,0,0};`
`177`	`177`	`ASSERT_EQ(leaf_index_expected, leaf_index_preds);`
`178`	`178`	`}`