Function TEST

tests/cpp/data/test_file_iterator.cc:16–44

namespace xgboost::data {
TEST(FileIterator, Basic) {
  auto check_n_features = [](FileIterator *iter) {
    size_t n_features = 0;
    iter->Reset();
    while (iter->Next()) {
      auto proxy = MakeProxy(iter->Proxy());
      auto csr = std::any_cast<std::shared_ptr<CSRArrayAdapter>>(proxy->Adapter());
      n_features = std::max(n_features, csr->NumColumns());
    }
    ASSERT_EQ(n_features, 5);
  };

  common::TemporaryDirectory tmpdir;
  {
    auto zpath = tmpdir.Str() + "/0-based.svm";
    CreateBigTestData(zpath, 3 * 64, true);
    zpath += "?indexing_mode=0&format=libsvm";
    FileIterator iter{zpath, 0, 1};
    check_n_features(&iter);
  }

  {
    auto opath = tmpdir.Str() + "/1-based.svm";
    CreateBigTestData(opath, 3 * 64, false);
    opath += "?indexing_mode=1&format=libsvm";
    FileIterator iter{opath, 0, 1};
    check_n_features(&iter);
  }
}
}  // namespace xgboost::data