cf2d86a4f6
* Add travis sanitizers tests. * Add gcc-7 in Travis. * Add SANITIZER_PATH for CMake. * Enable sanitizer tests in Travis. * Fix memory leaks in tests. * Fix all memory leaks reported by Address Sanitizer. * tests/cpp/helpers.h/CreateDMatrix now returns raw pointer.
36 lines
1.3 KiB
C++
36 lines
1.3 KiB
C++
// Copyright by Contributors
|
|
#include <xgboost/data.h>
|
|
#include "../../../src/data/simple_csr_source.h"
|
|
|
|
#include "../helpers.h"
|
|
|
|
TEST(SimpleCSRSource, SaveLoadBinary) {
|
|
std::string tmp_file = CreateSimpleTestData();
|
|
xgboost::DMatrix * dmat = xgboost::DMatrix::Load(tmp_file, true, false);
|
|
std::remove(tmp_file.c_str());
|
|
|
|
std::string tmp_binfile = TempFileName();
|
|
dmat->SaveToLocalFile(tmp_binfile);
|
|
xgboost::DMatrix * dmat_read = xgboost::DMatrix::Load(tmp_binfile, true, false);
|
|
std::remove(tmp_binfile.c_str());
|
|
|
|
EXPECT_EQ(dmat->Info().num_col_, dmat_read->Info().num_col_);
|
|
EXPECT_EQ(dmat->Info().num_row_, dmat_read->Info().num_row_);
|
|
EXPECT_EQ(dmat->Info().num_row_, dmat_read->Info().num_row_);
|
|
|
|
auto row_iter = dmat->RowIterator();
|
|
auto row_iter_read = dmat_read->RowIterator();
|
|
// Test the data read into the first row
|
|
row_iter->BeforeFirst(); row_iter->Next();
|
|
row_iter_read->BeforeFirst(); row_iter_read->Next();
|
|
auto first_row = row_iter->Value()[0];
|
|
auto first_row_read = row_iter_read->Value()[0];
|
|
EXPECT_EQ(first_row.size(), first_row_read.size());
|
|
EXPECT_EQ(first_row[2].index, first_row_read[2].index);
|
|
EXPECT_EQ(first_row[2].fvalue, first_row_read[2].fvalue);
|
|
row_iter = nullptr; row_iter_read = nullptr;
|
|
|
|
delete dmat;
|
|
delete dmat_read;
|
|
}
|