Commit 5ebd69f2 authored by Lukas Cejka's avatar Lukas Cejka Committed by Tomáš Oberhuber
Browse files

Added test for reset function.

parent c245151d
Loading
Loading
Loading
Loading
+43 −0
Original line number Diff line number Diff line
@@ -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"