f100b8d878
* Simplify DropTrees calling logic * Add `training` parameter for prediction method. * [Breaking]: Add `training` to C API. * Change for R and Python custom objective. * Correct comment. Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu> Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
#include <map>
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <xgboost/logging.h>
|
|
|
|
namespace xgboost {
|
|
|
|
TEST(Logging, Basic) {
|
|
std::map<std::string, std::string> args {};
|
|
std::string output;
|
|
|
|
args["verbosity"] = "0"; // silent
|
|
ConsoleLogger::Configure({args.cbegin(), args.cend()});
|
|
testing::internal::CaptureStderr();
|
|
LOG(DEBUG) << "Test silent.";
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_EQ(output.length(), 0);
|
|
|
|
args["verbosity"] = "3"; // debug
|
|
ConsoleLogger::Configure({args.cbegin(), args.cend()});
|
|
|
|
testing::internal::CaptureStderr();
|
|
LOG(WARNING) << "Test Log Warning.";
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_NE(output.find("WARNING"), std::string::npos);
|
|
|
|
testing::internal::CaptureStderr();
|
|
LOG(INFO) << "Test Log Info.";
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_NE(output.find("Test Log Info"), std::string::npos);
|
|
|
|
testing::internal::CaptureStderr();
|
|
LOG(DEBUG) << "Test Log Debug.";
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_NE(output.find("DEBUG"), std::string::npos);
|
|
|
|
args["verbosity"] = "1"; // warning
|
|
ConsoleLogger::Configure({args.cbegin(), args.cend()});
|
|
testing::internal::CaptureStderr();
|
|
LOG(INFO) << "INFO should not be displayed when set to warning.";
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_EQ(output.size(), 0);
|
|
|
|
args["silent"] = "True";
|
|
ConsoleLogger::Configure({args.cbegin(), args.cend()});
|
|
testing::internal::CaptureStderr();
|
|
LOG(INFO) << "Test silent parameter.";
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_EQ(output.length(), 0);
|
|
|
|
testing::internal::CaptureStderr();
|
|
LOG(CONSOLE) << "Test Log Console"; // ignore global setting.
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_NE(output.find("Test Log Console"), std::string::npos);
|
|
|
|
args["silent"] = "False";
|
|
args["verbosity"] = "2"; // restore
|
|
ConsoleLogger::Configure({args.cbegin(), args.cend()});
|
|
}
|
|
|
|
} // namespace xgboost
|