Commit 3d4c4c49 authored by Lukas Cejka's avatar Lukas Cejka
Browse files

Added test for setCompressedRowLengths function. This test is correct 

assuming that getRowLength() works correctly and doesn't need to be
tested.
parent c7d3de03

src/UnitTests/Matrices/SparseMatrixTest.h

+43 −1
Original line number Diff line number Diff line
@@ -58,6 +58,36 @@ void testSetDimensions() 
    EXPECT_EQ( m.getColumns(), 8);

}



template< typename Matrix >

void testSetCompressedRowLengths()

{

    Matrix m;

    const int rows = 10;

    const int cols = 11;

    

    m.reset();

    m.setDimensions( rows, cols );

    typename Matrix::CompressedRowLengthsVector rowLengths;

    rowLengths.setSize( rows );

    rowLengths.setValue( 3 );

    int value = 1;

    for (int i = 2; i < rows; i++)

        rowLengths.setElement( i, value++ );

    

    m.setCompressedRowLengths( rowLengths );

    

    EXPECT_EQ( m.getRowLength( 0), 3 );

    EXPECT_EQ( m.getRowLength( 1), 3 );

    EXPECT_EQ( m.getRowLength( 2), 1 );

    EXPECT_EQ( m.getRowLength( 3), 2 );

    EXPECT_EQ( m.getRowLength( 4), 3 );

    EXPECT_EQ( m.getRowLength( 5), 4 );

    EXPECT_EQ( m.getRowLength( 6), 5 );

    EXPECT_EQ( m.getRowLength( 7), 6 );

    EXPECT_EQ( m.getRowLength( 8), 7 );

    EXPECT_EQ( m.getRowLength( 9), 8 );

}



TEST( SparseMatrixTest, CSR_GetTypeTest_Host )

{

   host_testGetType< CSR_host_float, CSR_host_int >();
@@ -70,7 +100,7 @@ TEST( SparseMatrixTest, CSR_GetTypeTest_Cuda ) 
}

#endif



TEST( SparseMatrixTest, CSR_testSetDimensions_Host )

TEST( SparseMatrixTest, CSR_SetDimensionsTest_Host )

{

   testSetDimensions< CSR_host_int >();

}
@@ -82,6 +112,18 @@ TEST( SparseMatrixTest, CSR_SetDimensionsTest_Cuda ) 
}

#endif



TEST( SparseMatrixTest, CSR_setCompressedRowLengthsTest_Host )

{

   testSetCompressedRowLengths< CSR_host_int >();

}



#ifdef HAVE_CUDA

TEST( SparseMatrixTest, CSR_setCompressedRowLengthsTest_Cuda )

{

   testSetCompressedRowLengths< CSR_cuda_int >();

}

#endif



#endif



#include "../GtestMissingError.h"