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.
55 lines
1.5 KiB
C++
55 lines
1.5 KiB
C++
#include "../../../src/common/column_matrix.h"
|
|
#include "../helpers.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace xgboost {
|
|
namespace common {
|
|
TEST(DenseColumn, Test) {
|
|
auto dmat = CreateDMatrix(100, 10, 0.0);
|
|
GHistIndexMatrix gmat;
|
|
gmat.Init((*dmat).get(), 256);
|
|
ColumnMatrix column_matrix;
|
|
column_matrix.Init(gmat, 0.2);
|
|
|
|
for (auto i = 0ull; i < (*dmat)->Info().num_row_; i++) {
|
|
for (auto j = 0ull; j < (*dmat)->Info().num_col_; j++) {
|
|
auto col = column_matrix.GetColumn(j);
|
|
EXPECT_EQ(gmat.index[i * (*dmat)->Info().num_col_ + j],
|
|
col.GetGlobalBinIdx(i));
|
|
}
|
|
}
|
|
delete dmat;
|
|
}
|
|
|
|
TEST(SparseColumn, Test) {
|
|
auto dmat = CreateDMatrix(100, 1, 0.85);
|
|
GHistIndexMatrix gmat;
|
|
gmat.Init((*dmat).get(), 256);
|
|
ColumnMatrix column_matrix;
|
|
column_matrix.Init(gmat, 0.5);
|
|
auto col = column_matrix.GetColumn(0);
|
|
ASSERT_EQ(col.Size(), gmat.index.size());
|
|
for (auto i = 0ull; i < col.Size(); i++) {
|
|
EXPECT_EQ(gmat.index[gmat.row_ptr[col.GetRowIdx(i)]],
|
|
col.GetGlobalBinIdx(i));
|
|
}
|
|
delete dmat;
|
|
}
|
|
|
|
TEST(DenseColumnWithMissing, Test) {
|
|
auto dmat = CreateDMatrix(100, 1, 0.5);
|
|
GHistIndexMatrix gmat;
|
|
gmat.Init((*dmat).get(), 256);
|
|
ColumnMatrix column_matrix;
|
|
column_matrix.Init(gmat, 0.2);
|
|
auto col = column_matrix.GetColumn(0);
|
|
for (auto i = 0ull; i < col.Size(); i++) {
|
|
if (col.IsMissing(i)) continue;
|
|
EXPECT_EQ(gmat.index[gmat.row_ptr[col.GetRowIdx(i)]],
|
|
col.GetGlobalBinIdx(i));
|
|
}
|
|
delete dmat;
|
|
}
|
|
} // namespace common
|
|
} // namespace xgboost
|