c8c7b9649c
* Fix logging switch statement. * Remove debug_verbose_ in AllReducer. * Don't construct the stream when not needed. * Make default constructor deleted. * Remove redundant IsVerbose.
34 lines
1001 B
C++
34 lines
1001 B
C++
#include <gtest/gtest.h>
|
|
#include <xgboost/logging.h>
|
|
#include <string>
|
|
#include "../../../src/common/timer.h"
|
|
|
|
namespace xgboost {
|
|
namespace common {
|
|
TEST(Monitor, Logging) {
|
|
auto run_monitor =
|
|
[]() {
|
|
Monitor monitor_;
|
|
monitor_.Init("Monitor test");
|
|
monitor_.Start("basic");
|
|
monitor_.Stop("basic");
|
|
};
|
|
|
|
std::map<std::string, std::string> args = {std::make_pair("verbosity", "3")};
|
|
ConsoleLogger::Configure(args.cbegin(), args.cend());
|
|
testing::internal::CaptureStderr();
|
|
run_monitor();
|
|
std::string output = testing::internal::GetCapturedStderr();
|
|
ASSERT_NE(output.find("Monitor"), std::string::npos);
|
|
|
|
// Monitor only prints messages when set to DEBUG.
|
|
args = {std::make_pair("verbosity", "2")};
|
|
ConsoleLogger::Configure(args.cbegin(), args.cend());
|
|
testing::internal::CaptureStderr();
|
|
run_monitor();
|
|
output = testing::internal::GetCapturedStderr();
|
|
ASSERT_EQ(output.size(), 0);
|
|
}
|
|
} // namespace common
|
|
} // namespace xgboost
|