#include <sparse_matrix_test.h>

Inheritance diagram for SparseMatrixTest< DerivedClass >:

Public Member Functions
void	setUp ()

void	tearDown ()

void	setValues ()

void	testValues ()

void	testGetAndSet ()

void	testReadMatlab (const std::string &filename)

void	testReadMatlab1 ()

void	testReadMatlab2 ()

void	testReadMatlab4 ()

void	testWriteAndRead ()

void	testClone ()

Protected Attributes
std::string	libmesh_suite_name

libMesh::Parallel::Communicator *	my_comm = nullptr

std::unique_ptr< DerivedClass >	matrix

libMesh::numeric_index_type	nonsquare = 1

libMesh::numeric_index_type	local_m

libMesh::numeric_index_type	local_n

libMesh::numeric_index_type	global_m

libMesh::numeric_index_type	global_n

const libMesh::Real	_tolerance = libMesh::TOLERANCE * libMesh::TOLERANCE

Detailed Description

template<class DerivedClass>
class SparseMatrixTest< DerivedClass >

Definition at line 33 of file sparse_matrix_test.h.

Member Function Documentation

◆ setUp()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::setUp ( )

inline

Definition at line 37 of file sparse_matrix_test.h.

   {
     // By default we'll use the whole communicator in parallel;
     // Serial-only NumericVector subclasses will need to override
     // this and set something else first.
     if (!my_comm)
       my_comm = TestCommWorld;
 
     matrix = std::make_unique<DerivedClass>(*my_comm);
 
     // Use the same even partitioning that we'll auto-deduce in matrix
     // read, to make it easier to test parallel reads
 
     global_m = global_n = my_comm->size() * 5.75 - 1;
     if (nonsquare)
       global_n *= 1.3;
 
     libMesh::dof_id_type
       row_start =     my_comm->rank() * global_m / my_comm->size(),
       row_stop  = (my_comm->rank()+1) * global_m / my_comm->size();
     libMesh::dof_id_type
       col_start =     my_comm->rank() * global_n / my_comm->size(),
       col_stop  = (my_comm->rank()+1) * global_n / my_comm->size();
 
     local_m = row_stop - row_start;
     local_n = col_stop - col_start;
 
     // Let's just play around with locally dense blocks for now
     matrix->init(global_m,
                  global_n,
                  local_m,
                  local_n,
                  /*nnz=*/local_n,
                  /*noz=*/5);
   }

◆ setValues()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::setValues ( )

inline

Definition at line 77 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::testClone(), SparseMatrixTest< LaspackMatrix< Number > >::testGetAndSet(), and SparseMatrixTest< LaspackMatrix< Number > >::testWriteAndRead().

   {
     std::vector<libMesh::numeric_index_type> rows(local_m);
     std::iota(rows.begin(), rows.end(), matrix->row_start());
     std::vector<libMesh::numeric_index_type> cols(local_n);
     std::iota(cols.begin(), cols.end(), matrix->col_start());
 
     libMesh::DenseMatrix<libMesh::Number> local(local_m, local_n);
 
     for (auto i : libMesh::make_range(local_m))
       for (auto j : libMesh::make_range(local_n))
         local(i, j) = (i + 1) * (j + 1) * (my_comm->rank() + 1);
 
     matrix->zero();
 
     matrix->add_matrix(local, rows, cols);
     matrix->close();
   }

◆ tearDown()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::tearDown ( )

inline

Definition at line 74 of file sparse_matrix_test.h.

74 {}

◆ testClone()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testClone ( )

inline

Definition at line 223 of file sparse_matrix_test.h.

   {
     LOG_UNIT_TEST;
 
     setValues();
 
     // Matrix must be closed before it can be cloned.
     matrix->close();
 
     {
       // Create copy, test that it can go out of scope
       auto copy = matrix->clone();
 
       // Check that matrices have the same local/global sizes
       CPPUNIT_ASSERT_EQUAL(copy->m(), matrix->m());
       CPPUNIT_ASSERT_EQUAL(copy->n(), matrix->n());
       CPPUNIT_ASSERT_EQUAL(copy->local_m(), matrix->local_m());
       CPPUNIT_ASSERT_EQUAL(copy->row_start(), matrix->row_start());
       CPPUNIT_ASSERT_EQUAL(copy->row_stop(), matrix->row_stop());
 
       // Check that copy has same values as original
       LIBMESH_ASSERT_FP_EQUAL(copy->l1_norm(), matrix->l1_norm(), _tolerance);
       CPPUNIT_ASSERT(relative_fuzzy_equals(*matrix, *copy));
       copy->scale(2);
       CPPUNIT_ASSERT(!relative_fuzzy_equals(*matrix, *copy));
     }
 
     {
       // Create zero copy
       auto zero_copy = matrix->zero_clone();
 
       // Check that matrices have the same local/global sizes
       CPPUNIT_ASSERT_EQUAL(zero_copy->m(), matrix->m());
       CPPUNIT_ASSERT_EQUAL(zero_copy->n(), matrix->n());
       CPPUNIT_ASSERT_EQUAL(zero_copy->local_m(), matrix->local_m());
       CPPUNIT_ASSERT_EQUAL(zero_copy->row_start(), matrix->row_start());
       CPPUNIT_ASSERT_EQUAL(zero_copy->row_stop(), matrix->row_stop());
 
       // Check that zero_copy has same values as original
       LIBMESH_ASSERT_FP_EQUAL(0.0, zero_copy->l1_norm(), _tolerance);
     }
   }

◆ testGetAndSet()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testGetAndSet ( )

inline

Definition at line 137 of file sparse_matrix_test.h.

   {
     LOG_UNIT_TEST;
 
     setValues();
 
     testValues();
   }

◆ testReadMatlab()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testReadMatlab ( const std::string & filename )

inline

Definition at line 146 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::testReadMatlab1(), SparseMatrixTest< LaspackMatrix< Number > >::testReadMatlab2(), and SparseMatrixTest< LaspackMatrix< Number > >::testReadMatlab4().

   {
     // Laspack doesn't handle non-square matrices)
     if (matrix->solver_package() == libMesh::LASPACK_SOLVERS)
       return;
 
     matrix->clear();
 
     auto matrix2 = std::make_unique<DerivedClass>(*my_comm);
 
     matrix->read(filename);
 
 #ifndef LIBMESH_HAVE_GZSTREAM
     return;
 #endif
 
     matrix2->read(std::string(filename)+".gz");
 
     // We need some more SparseMatrix operators, but not today
     CPPUNIT_ASSERT(matrix->l1_norm() == matrix2->l1_norm());
     CPPUNIT_ASSERT(matrix->linfty_norm() == matrix2->linfty_norm());
   }

◆ testReadMatlab1()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testReadMatlab1 ( )

inline

Definition at line 170 of file sparse_matrix_test.h.

   {
     LOG_UNIT_TEST;
     testReadMatlab("matrices/geom_1_extraction_op.m");
   }

◆ testReadMatlab2()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testReadMatlab2 ( )

inline

Definition at line 177 of file sparse_matrix_test.h.

   {
     LOG_UNIT_TEST;
     testReadMatlab("matrices/geom_2_extraction_op.m");
   }

◆ testReadMatlab4()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testReadMatlab4 ( )

inline

Definition at line 184 of file sparse_matrix_test.h.

   {
     LOG_UNIT_TEST;
     testReadMatlab("matrices/geom_4_extraction_op.m");
   }

◆ testValues()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testValues ( )

inline

Definition at line 97 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::testGetAndSet(), and SparseMatrixTest< LaspackMatrix< Number > >::testWriteAndRead().

   {
     auto functor = [this]()
     {
       std::vector<libMesh::numeric_index_type> cols_to_get;
       std::vector<libMesh::Number> values;
       const libMesh::numeric_index_type col_start =
         matrix->col_start();
       for (libMesh::numeric_index_type i :
            libMesh::make_range(matrix->row_start(),
                                matrix->row_stop()))
         {
           matrix->get_row(i, cols_to_get, values);
           for (libMesh::numeric_index_type col_j :
                libMesh::index_range(cols_to_get))
             {
               CPPUNIT_ASSERT_EQUAL(cols_to_get[col_j], col_start + col_j);
               LIBMESH_ASSERT_NUMBERS_EQUAL
                 ((i - matrix->row_start() + 1) * (col_j + 1) * (my_comm->rank() + 1),
                  values[col_j], _tolerance);
             }
         }
     };
 
 #ifdef LIBMESH_HAVE_CXX11_THREAD
     auto num_threads = std::min(unsigned(2),
                                 std::max(
                                   std::thread::hardware_concurrency(),
                                   unsigned(1)));
     std::vector<std::thread> threads(num_threads);
     for (unsigned int thread = 0; thread < num_threads; ++thread)
       threads[thread] = std::thread(functor);
     std::for_each(threads.begin(), threads.end(),
                   [](std::thread & x){x.join();});
 #else
     functor();
 #endif
   }

◆ testWriteAndRead()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testWriteAndRead ( )

inline

Definition at line 191 of file sparse_matrix_test.h.

   {
     LOG_UNIT_TEST;
 
     setValues();
 
     // If we're working with serial matrices then just print one of
     // them so they don't step on the others' toes.
     if (matrix->n_processors() > 1 ||
         TestCommWorld->rank() == 0)
       matrix->print_matlab(libmesh_suite_name+"_matrix.m");
 
     matrix->clear();
 
     // Let's make sure we don't have any race conditions; we have
     // multiple SparseMatrix subclasses that might be trying to read
     // and write the same file.
 
     TestCommWorld->barrier();
 
 #ifdef LIBMESH_USE_COMPLEX_NUMBERS
     // We're not supporting complex reads quite yet
     return;
 #endif
 
     matrix->read(libmesh_suite_name+"_matrix.m");
 
     TestCommWorld->barrier();
 
     testValues();
   }