xgboost_lib-sys 3.0.5

Native bindings to the xgboost library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

#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);

  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["verbosity"] = "2";  // restore
  ConsoleLogger::Configure({args.cbegin(), args.cend()});
}

}  // namespace xgboost