TEST(Matrix, CpuSparseMatrixRandUniform) {
const size_t HEIGHT = 5;
const size_t WIDTH = 10;
const size_t NNZ = HEIGHT * WIDTH;
int* major = nullptr;
int* minor = nullptr;
size_t majorLen = 0;
size_t minorLen = 0;
size_t nnz = 0;
for (auto valueType : {NO_VALUE, FLOAT_VALUE}) {
for (auto format : {SPARSE_CSR, SPARSE_CSC}) {
CpuSparseMatrixPtr matA = std::make_shared<CpuSparseMatrix>(
HEIGHT, WIDTH, size_t(NNZ * 0.1), valueType, format);
matA->randomizeUniform();
nnz = matA->getElementCnt();
if (format == SPARSE_CSR) {
majorLen = matA->getHeight() + 1;
minorLen = matA->getElementCnt();
major = matA->getRows();
minor = matA->getCols();
} else {
majorLen = matA->getWidth() + 1;
minorLen = matA->getElementCnt();
major = matA->getCols();
minor = matA->getRows();
}
sparseValid(major, minor, nnz, majorLen, minorLen);
}
}
}
TEST(Matrix, CpuSparseMatrixSubMatrix) {
const size_t HEIGHT = 10;
const size_t WIDTH = 10;
const size_t NNZ = HEIGHT * WIDTH;
for (auto valueType : {FLOAT_VALUE, NO_VALUE}) {
size_t startRow = 3;
size_t rowNum = 2;
real sparseRate = 0.1;
/*sparse matrix init and get subMatrix*/
CpuSparseMatrixPtr matA = std::make_shared<CpuSparseMatrix>(
HEIGHT, WIDTH, size_t(NNZ * sparseRate), valueType, SPARSE_CSR);
matA->randomizeUniform();
CpuSparseMatrixPtr matB = std::dynamic_pointer_cast<CpuSparseMatrix>(
matA->subMatrix(startRow, rowNum));
int start = matA->getRows()[startRow];
int end = matA->getRows()[startRow + rowNum];
/*compare two matrix*/
ASSERT_EQ(matB->getElementCnt(), size_t(end - start));
if (valueType == FLOAT_VALUE) {
for (size_t i = 0; i < matB->getElementCnt(); i++) {
ASSERT_FLOAT_EQ(matB->getValue()[start + i],
matA->getValue()[start + i]);
}
}
for (size_t i = 0; i < matB->getElementCnt(); i++) {
ASSERT_EQ(matB->getCols()[start + i], matA->getCols()[start + i]);
}
for (size_t i = 0; i < rowNum; i++) {
ASSERT_EQ(matB->getRows()[i], matA->getRows()[startRow + i]);
}
}
}