Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 326440ae authored by hhakim's avatar hhakim
Browse files

Add a unit test for MatButterfly::getNonZeros and getNBytes().

parent 1804fc40
No related branches found
No related tags found
No related merge requests found
......@@ -5,6 +5,7 @@
typedef @TEST_FPP@ FPP;
using namespace Faust;
using namespace std;
void testClone(MatButterfly<FPP, Cpu>& butterflyMat)
{
......@@ -97,6 +98,19 @@ void testTranspose(MatButterfly<FPP, Cpu>& butterflyMat, MatSparse<FPP, Cpu>& sp
std::cout << "MatButterfly transpose OK" << std::endl;
}
void testNNZSize(MatButterfly<FPP, Cpu>& butterflyMat, MatSparse<FPP, Cpu>& spButterflyMat)
{
assert(butterflyMat.getNonZeros() == spButterflyMat.getNonZeros());
// cout << "nbytes sp, butter:" << spButterflyMat.getNBytes() << " " << butterflyMat.getNBytes() << endl;
assert(butterflyMat.getNBytes() == (butterflyMat.getNbRow() * (2 * sizeof(FPP) + sizeof(int))));
MatButterfly<FPP, Cpu> tbutterflyMat(butterflyMat);
tbutterflyMat.transpose();
// cout << "nbytes sp, tbutter:" << spButterflyMat.getNBytes() << " " << tbutterflyMat.getNBytes() << endl;
assert(tbutterflyMat.getNBytes() == (tbutterflyMat.getNbRow() * (3 * sizeof(FPP) + sizeof(int))));
std::cout << "MatButterfly getNonZeros() and getNBytes() OK" << std::endl;
}
int main(int argc, char** argv)
{
int log2size = 4;
......@@ -159,6 +173,8 @@ int main(int argc, char** argv)
testClone(butterflyMat);
testTranspose(butterflyMat, spButterflyMat);
testNNZSize(butterflyMat, spButterflyMat);
return EXIT_SUCCESS;
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment