SparseMatrixTest< DerivedClass > Class Template Reference

#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	testReadHDF5 ()
void	testTransposeNorms ()
void	testWriteAndRead (const std::string &filename)
void	testWriteAndReadHDF5 ()
void	testWriteAndReadMatlab ()
void	testWriteAndReadZippedMatlab ()
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 49 of file sparse_matrix_test.h.

Member Function Documentation

setUp()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::setUp ( )

inline

Definition at line 53 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 93 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::testClone(), SparseMatrixTest< LaspackMatrix< Number > >::testGetAndSet(), SparseMatrixTest< LaspackMatrix< Number > >::testTransposeNorms(), 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 90 of file sparse_matrix_test.h.

{}

testClone()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testClone ( )

inline

Definition at line 296 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 153 of file sparse_matrix_test.h.

  {
    LOG_UNIT_TEST;

    setValues();

    testValues();
  }

testReadHDF5()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testReadHDF5 ( )

inline

Definition at line 209 of file sparse_matrix_test.h.

  {
    LOG_UNIT_TEST;

    matrix->clear();
    auto matrix2 = std::make_unique<DerivedClass>(*my_comm);
    matrix->read("matrices/geom_1_extraction_op.m");
    matrix2->read("matrices/geom_1_extraction_op.h5");

    // 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());
  }

testReadMatlab()

template<class DerivedClass>

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

inline

Definition at line 163 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 187 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 194 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 201 of file sparse_matrix_test.h.

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

testTransposeNorms()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testTransposeNorms ( )

inline

Definition at line 225 of file sparse_matrix_test.h.

  {
    LOG_UNIT_TEST;

    setValues();

    auto matrix2 = std::make_unique<DerivedClass>(*my_comm);
    matrix->get_transpose(*matrix2);
    LIBMESH_ASSERT_FP_EQUAL(matrix->l1_norm(), matrix2->linfty_norm(), _tolerance);
    LIBMESH_ASSERT_FP_EQUAL(matrix2->l1_norm(), matrix->linfty_norm(), _tolerance);
  }

testValues()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testValues ( )

inline

Definition at line 113 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 ( const std::string &  filename )

inline

Definition at line 238 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::testWriteAndReadHDF5(), SparseMatrixTest< LaspackMatrix< Number > >::testWriteAndReadMatlab(), and SparseMatrixTest< LaspackMatrix< Number > >::testWriteAndReadZippedMatlab().

  {
    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(filename);

    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(filename);

    TestCommWorld->barrier();

    testValues();
  }

testWriteAndReadHDF5()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testWriteAndReadHDF5 ( )

inline

Definition at line 271 of file sparse_matrix_test.h.

  {
    // This capability is not yet supported when libmesh is compiled
    // with complex number support
#ifndef LIBMESH_USE_COMPLEX_NUMBERS
    testWriteAndRead("M.h5");
#endif
  }

testWriteAndReadMatlab()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testWriteAndReadMatlab ( )

inline

Definition at line 282 of file sparse_matrix_test.h.

  {
    // Use a very short filename, because we had a bug with that and
    // we want to test it.
    testWriteAndRead("M.m");
  }

testWriteAndReadZippedMatlab()

template<class DerivedClass>

void SparseMatrixTest< DerivedClass >::testWriteAndReadZippedMatlab ( )

inline

Definition at line 289 of file sparse_matrix_test.h.

  {
#ifdef LIBMESH_HAVE_GZSTREAM
    testWriteAndRead("Mzipped.m.gz");
#endif
  }

Member Data Documentation

_tolerance

template<class DerivedClass>

const libMesh::Real SparseMatrixTest< DerivedClass >::_tolerance = libMesh::TOLERANCE * libMesh::TOLERANCE

protected

Definition at line 351 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::testClone(), SparseMatrixTest< LaspackMatrix< Number > >::testTransposeNorms(), and SparseMatrixTest< LaspackMatrix< Number > >::testValues().

global_m

template<class DerivedClass>

libMesh::numeric_index_type SparseMatrixTest< DerivedClass >::global_m

protected

Definition at line 347 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp().

global_n

template<class DerivedClass>

libMesh::numeric_index_type SparseMatrixTest< DerivedClass >::global_n

protected

Definition at line 347 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp().

libmesh_suite_name

template<class DerivedClass>

std::string SparseMatrixTest< DerivedClass >::libmesh_suite_name

protected

Definition at line 341 of file sparse_matrix_test.h.

local_m

template<class DerivedClass>

libMesh::numeric_index_type SparseMatrixTest< DerivedClass >::local_m

protected

Definition at line 347 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp(), and SparseMatrixTest< LaspackMatrix< Number > >::setValues().

local_n

template<class DerivedClass>

libMesh::numeric_index_type SparseMatrixTest< DerivedClass >::local_n

protected

Definition at line 347 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp(), and SparseMatrixTest< LaspackMatrix< Number > >::setValues().

matrix

template<class DerivedClass>

std::unique_ptr<DerivedClass> SparseMatrixTest< DerivedClass >::matrix

protected

Definition at line 345 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp(), SparseMatrixTest< LaspackMatrix< Number > >::setValues(), SparseMatrixTest< LaspackMatrix< Number > >::testClone(), SparseMatrixTest< LaspackMatrix< Number > >::testReadHDF5(), SparseMatrixTest< LaspackMatrix< Number > >::testReadMatlab(), SparseMatrixTest< LaspackMatrix< Number > >::testTransposeNorms(), SparseMatrixTest< LaspackMatrix< Number > >::testValues(), and SparseMatrixTest< LaspackMatrix< Number > >::testWriteAndRead().

my_comm

template<class DerivedClass>

libMesh::Parallel::Communicator* SparseMatrixTest< DerivedClass >::my_comm = nullptr

protected

Definition at line 343 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp(), SparseMatrixTest< LaspackMatrix< Number > >::setValues(), and SparseMatrixTest< LaspackMatrix< Number > >::testValues().

nonsquare

template<class DerivedClass>

libMesh::numeric_index_type SparseMatrixTest< DerivedClass >::nonsquare = 1

protected

Definition at line 347 of file sparse_matrix_test.h.

Referenced by SparseMatrixTest< LaspackMatrix< Number > >::setUp().

---

The documentation for this class was generated from the following file:

• tests/numerics/sparse_matrix_test.h