08ce495b5d
- Pass context from booster to DMatrix. - Use context instead of integer for `n_threads`. - Check the consistency configuration for `max_bin`. - Test for all combinations of initialization options.
64 lines
2.2 KiB
C++
64 lines
2.2 KiB
C++
/**
|
|
* Copyright 2021-2023, XGBoost contributors
|
|
*/
|
|
#include <gtest/gtest.h>
|
|
#include <xgboost/data.h> // for CSCPage, SortedCSCPage, SparsePage
|
|
|
|
#include <memory> // for allocator, unique_ptr, __shared_ptr_ac...
|
|
#include <string> // for char_traits, operator+, basic_string
|
|
|
|
#include "../../../src/data/sparse_page_writer.h" // for CreatePageFormat
|
|
#include "../helpers.h" // for RandomDataGenerator
|
|
#include "dmlc/filesystem.h" // for TemporaryDirectory
|
|
#include "dmlc/io.h" // for SeekStream, Stream
|
|
#include "gtest/gtest_pred_impl.h" // for Test, AssertionResult, ASSERT_EQ, TEST
|
|
#include "xgboost/context.h" // for Context
|
|
|
|
namespace xgboost {
|
|
namespace data {
|
|
template <typename S> void TestSparsePageRawFormat() {
|
|
std::unique_ptr<SparsePageFormat<S>> format{CreatePageFormat<S>("raw")};
|
|
Context ctx;
|
|
|
|
auto m = RandomDataGenerator{100, 14, 0.5}.GenerateDMatrix();
|
|
ASSERT_TRUE(m->SingleColBlock());
|
|
dmlc::TemporaryDirectory tmpdir;
|
|
std::string path = tmpdir.path + "/sparse.page";
|
|
S orig;
|
|
{
|
|
// block code to flush the stream
|
|
std::unique_ptr<dmlc::Stream> fo{dmlc::Stream::Create(path.c_str(), "w")};
|
|
for (auto const &page : m->GetBatches<S>(&ctx)) {
|
|
orig.Push(page);
|
|
format->Write(page, fo.get());
|
|
}
|
|
}
|
|
|
|
S page;
|
|
std::unique_ptr<dmlc::SeekStream> fi{dmlc::SeekStream::CreateForRead(path.c_str())};
|
|
format->Read(&page, fi.get());
|
|
for (size_t i = 0; i < orig.data.Size(); ++i) {
|
|
ASSERT_EQ(page.data.HostVector()[i].fvalue,
|
|
orig.data.HostVector()[i].fvalue);
|
|
ASSERT_EQ(page.data.HostVector()[i].index, orig.data.HostVector()[i].index);
|
|
}
|
|
for (size_t i = 0; i < orig.offset.Size(); ++i) {
|
|
ASSERT_EQ(page.offset.HostVector()[i], orig.offset.HostVector()[i]);
|
|
}
|
|
ASSERT_EQ(page.base_rowid, orig.base_rowid);
|
|
}
|
|
|
|
TEST(SparsePageRawFormat, SparsePage) {
|
|
TestSparsePageRawFormat<SparsePage>();
|
|
}
|
|
|
|
TEST(SparsePageRawFormat, CSCPage) {
|
|
TestSparsePageRawFormat<CSCPage>();
|
|
}
|
|
|
|
TEST(SparsePageRawFormat, SortedCSCPage) {
|
|
TestSparsePageRawFormat<SortedCSCPage>();
|
|
}
|
|
} // namespace data
|
|
} // namespace xgboost
|