From 283818560b95af9a1d1193a1b7b3f9de8257be4c Mon Sep 17 00:00:00 2001 From: Lukas Cejka <lukas.ostatek@gmail.com> Date: Sun, 4 Nov 2018 19:30:40 +0100 Subject: [PATCH] Added test for reset function. --- src/UnitTests/Matrices/SparseMatrixTest.h | 43 +++++++++++++++++++++++ 1 file changed, 43 insertions(+) diff --git a/src/UnitTests/Matrices/SparseMatrixTest.h b/src/UnitTests/Matrices/SparseMatrixTest.h index 5fde08cd5a..71134d82e3 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" -- GitLab