Add global configuration (#6414)
* Add management functions for global configuration: XGBSetGlobalConfig(), XGBGetGlobalConfig(). * Add Python interface: set_config(), get_config(), and config_context(). * Add unit tests for Python * Add R interface: xgb.set.config(), xgb.get.config() * Add unit tests for R Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
This commit is contained in:
committed by
GitHub
parent
c2ba4fb957
commit
fb56da5e8b
@@ -212,4 +212,50 @@ TEST(CAPI, Exception) {
|
||||
// Not null
|
||||
ASSERT_TRUE(error);
|
||||
}
|
||||
|
||||
TEST(CAPI, XGBGlobalConfig) {
|
||||
int ret;
|
||||
{
|
||||
const char *config_str = R"json(
|
||||
{
|
||||
"verbosity": 0
|
||||
}
|
||||
)json";
|
||||
ret = XGBSetGlobalConfig(config_str);
|
||||
ASSERT_EQ(ret, 0);
|
||||
const char *updated_config_cstr;
|
||||
ret = XGBGetGlobalConfig(&updated_config_cstr);
|
||||
ASSERT_EQ(ret, 0);
|
||||
|
||||
std::string updated_config_str{updated_config_cstr};
|
||||
auto updated_config =
|
||||
Json::Load({updated_config_str.data(), updated_config_str.size()});
|
||||
ASSERT_EQ(get<Integer>(updated_config["verbosity"]), 0);
|
||||
}
|
||||
{
|
||||
const char *config_str = R"json(
|
||||
{
|
||||
"foo": 0
|
||||
}
|
||||
)json";
|
||||
ret = XGBSetGlobalConfig(config_str);
|
||||
ASSERT_EQ(ret , -1);
|
||||
auto err = std::string{XGBGetLastError()};
|
||||
ASSERT_NE(err.find("foo"), std::string::npos);
|
||||
}
|
||||
{
|
||||
const char *config_str = R"json(
|
||||
{
|
||||
"foo": 0,
|
||||
"verbosity": 0
|
||||
}
|
||||
)json";
|
||||
ret = XGBSetGlobalConfig(config_str);
|
||||
ASSERT_EQ(ret , -1);
|
||||
auto err = std::string{XGBGetLastError()};
|
||||
ASSERT_NE(err.find("foo"), std::string::npos);
|
||||
ASSERT_EQ(err.find("verbosity"), std::string::npos);
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace xgboost
|
||||
|
||||
22
tests/cpp/test_global_config.cc
Normal file
22
tests/cpp/test_global_config.cc
Normal file
@@ -0,0 +1,22 @@
|
||||
#include <gtest/gtest.h>
|
||||
#include <xgboost/json.h>
|
||||
#include <xgboost/logging.h>
|
||||
#include <xgboost/global_config.h>
|
||||
|
||||
namespace xgboost {
|
||||
|
||||
TEST(GlobalConfiguration, Verbosity) {
|
||||
// Configure verbosity via global configuration
|
||||
Json config{JsonObject()};
|
||||
config["verbosity"] = String("0");
|
||||
auto& global_config = *GlobalConfigThreadLocalStore::Get();
|
||||
FromJson(config, &global_config);
|
||||
// Now verbosity should be updated
|
||||
EXPECT_EQ(ConsoleLogger::GlobalVerbosity(), ConsoleLogger::LogVerbosity::kSilent);
|
||||
EXPECT_NE(ConsoleLogger::LogVerbosity::kSilent, ConsoleLogger::DefaultVerbosity());
|
||||
// GetConfig() should also return updated verbosity
|
||||
Json current_config { ToJson(*GlobalConfigThreadLocalStore::Get()) };
|
||||
EXPECT_EQ(get<String>(current_config["verbosity"]), "0");
|
||||
}
|
||||
|
||||
} // namespace xgboost
|
||||
Reference in New Issue
Block a user