Unify test helpers for creating ctx. (#9274)

2023-06-10 03:35:22 +08:00
parent ea0deeca68
commit 152e2fb072
36 changed files with 161 additions and 169 deletions
--- a/tests/cpp/plugin/test_example_objective.cc
+++ b/tests/cpp/plugin/test_example_objective.cc
@@ -4,12 +4,10 @@
 #include "../helpers.h"

 namespace xgboost {
-
 TEST(Plugin, ExampleObjective) {
-  xgboost::Context ctx = CreateEmptyGenericParam(GPUIDX);
+  xgboost::Context ctx = MakeCUDACtx(GPUIDX);
  auto* obj = xgboost::ObjFunction::Create("mylogistic", &ctx);
  ASSERT_EQ(obj->DefaultEvalMetric(), std::string{"logloss"});
  delete obj;
 }
-
 }  // namespace xgboost