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