Function VerifyGetSetFeatureColumnSplit

tests/cpp/data/test_metainfo.cc:82–118

namespace {
void VerifyGetSetFeatureColumnSplit() {
  xgboost::MetaInfo info;
  info.data_split_mode = DataSplitMode::kCol;
  auto const world_size = collective::GetWorldSize();

  auto constexpr kCols{2};
  std::vector<std::string> types{u8"float", u8"c"};
  std::vector<char const *> c_types(kCols);
  std::transform(types.cbegin(), types.cend(), c_types.begin(),
                 [](auto const &str) { return str.c_str(); });
  info.num_col_ = kCols;
  ASSERT_THAT([&] { info.SetFeatureInfo(u8"feature_type", c_types.data(), c_types.size()); },
              GMockThrow("Length of feature_type must be equal to number of columns"));
  info.num_col_ = kCols * world_size;
  EXPECT_NO_THROW(info.SetFeatureInfo(u8"feature_type", c_types.data(), c_types.size()));
  std::vector<std::string> expected_type_names{u8"float", u8"c",     u8"float",
                                               u8"c",     u8"float", u8"c"};
  EXPECT_EQ(info.feature_type_names, expected_type_names);
  std::vector<xgboost::FeatureType> expected_types{
      xgboost::FeatureType::kNumerical, xgboost::FeatureType::kCategorical,
      xgboost::FeatureType::kNumerical, xgboost::FeatureType::kCategorical,
      xgboost::FeatureType::kNumerical, xgboost::FeatureType::kCategorical};
  EXPECT_EQ(info.feature_types.HostVector(), expected_types);

  std::vector<std::string> names{u8"feature0", u8"feature1"};
  std::vector<char const *> c_names(kCols);
  std::transform(names.cbegin(), names.cend(), c_names.begin(),
                 [](auto const &str) { return str.c_str(); });
  info.num_col_ = kCols;
  ASSERT_THAT([&] { info.SetFeatureInfo(u8"feature_name", c_names.data(), c_names.size()); },
              GMockThrow("Length of feature_name must be equal to number of columns"));
  info.num_col_ = kCols * world_size;
  EXPECT_NO_THROW(info.SetFeatureInfo(u8"feature_name", c_names.data(), c_names.size()));
  std::vector<std::string> expected_names{u8"0.feature0", u8"0.feature1", u8"1.feature0",
                                          u8"1.feature1", u8"2.feature0", u8"2.feature1"};
  EXPECT_EQ(info.feature_names, expected_names);
}
}  // anonymous namespace

TEST(MetaInfo, GetSetFeatureColumnSplit) {