diff --git a/src/UnitTests/Matrices/SparseMatrixTest.h b/src/UnitTests/Matrices/SparseMatrixTest.h index 5fde08cd5a32006e955fb11eb6bb08bbc0807ecf..71134d82e3b6fb76049cfc5e975c981ce93c11bf 100644 --- a/src/UnitTests/Matrices/SparseMatrixTest.h +++ b/src/UnitTests/Matrices/SparseMatrixTest.h @@ -108,6 +108,21 @@ void test_SetLike() EXPECT_EQ( m1.getColumns(), m2.getColumns() ); } +template< typename Matrix > +void test_Reset() +{ + const int rows = 5; + const int cols = 4; + + Matrix m; + m.setDimensions( rows, cols ); + + m.reset(); + + EXPECT_EQ( m.getRows(), 0 ); + EXPECT_EQ( m.getColumns(), 0 ); +} + TEST( SparseMatrixTest, CSR_GetTypeTest_Host ) { host_test_GetType< CSR_host_float, CSR_host_int >(); @@ -156,6 +171,34 @@ TEST( SparseMatrixTest, CSR_setLikeTest_Cuda ) } #endif +TEST( SparseMatrixTest, CSR_resetTest_Host ) +{ + { + SCOPED_TRACE( "CSR_resetTest_Host_Float" ); + test_Reset< CSR_host_float >(); + } + + { + SCOPED_TRACE( "CSR_resetTest_Host_Int" ); + test_Reset< CSR_host_int >(); + } +} + +#ifdef HAVE_CUDA +TEST( SparseMatrixTest, CSR_resetTest_Cuda ) +{ + { + SCOPED_TRACE( "CSR_resetTest_Cuda_Float" ); + test_Reset< CSR_cuda_float >(); + } + + { + SCOPED_TRACE( "CSR_resetTest_Cuda_Int" ); + test_Reset< CSR_cuda_int >(); + } +} +#endif + #endif #include "../GtestMissingError.h"