Function TestCatContainerStr

tests/cpp/data/test_cat_container.h:38–66

template <typename DfTest, typename EqCheck>
auto TestCatContainerStr(Context const* ctx, EqCheck&& is_eq) {
  auto df = DfTest::Make(DfTest::MakeStrs("abc", "bcd", "cde", "ab"));
  auto h_df = df.View();
  auto cats = test_cat_detail::FromDf(ctx, h_df);
  if (ctx->IsCPU()) {
    test_cat_detail::HostCheck(cats);
  } else {
    test_cat_detail::DeviceCheck(cats);
  }

  [&] {
    ASSERT_EQ(df.View().columns.size(), cats.NumFeatures());
  }();

  cats.Sort(ctx);

  auto sol = std::vector<bst_cat_t>{3, 0, 1, 2};
  auto sorted_idx = cats.RefSortedIndex(ctx);
  is_eq(sorted_idx, sol);
  [&] {
    auto view = cats.HostView();
    ASSERT_EQ(view.n_total_cats, sol.size());
    ASSERT_EQ(view.feature_segments.size(), 2ul);
    ASSERT_EQ(view.feature_segments[0], 0);
    ASSERT_EQ(view.feature_segments[1], static_cast<bst_cat_t>(sol.size()));
  }();

  return df;
}

template <typename DfTest, typename EqCheck>
auto TestCatContainerMixed(Context const* ctx, EqCheck&& is_eq) {