PetscMatrixTest Class Reference

Inheritance diagram for PetscMatrixTest:

Public Member Functions
	CPPUNIT_TEST_SUITE (PetscMatrixTest)
	CPPUNIT_TEST (testGetAndSet)
	CPPUNIT_TEST_SUITE_END ()
void	setUp ()
void	tearDown ()
void	testGetAndSet ()

Private Attributes
Parallel::Communicator *	_comm
std::unique_ptr< PetscMatrix< Number > >	_matrix
numeric_index_type	_local_size
numeric_index_type	_global_size
std::vector< numeric_index_type >	_i
const Real	_tolerance = TOLERANCE * TOLERANCE

Detailed Description

Definition at line 19 of file petsc_matrix_test.C.

Member Function Documentation

CPPUNIT_TEST()

PetscMatrixTest::CPPUNIT_TEST

(

testGetAndSet

)

CPPUNIT_TEST_SUITE()

PetscMatrixTest::CPPUNIT_TEST_SUITE

(

PetscMatrixTest

)

CPPUNIT_TEST_SUITE_END()

PetscMatrixTest::CPPUNIT_TEST_SUITE_END

(

)

setUp()

void PetscMatrixTest::setUp ( )

inline

Definition at line 29 of file petsc_matrix_test.C.

  {
    _comm = TestCommWorld;
    _matrix = libmesh_make_unique<PetscMatrix<Number>>(*_comm);
 
    numeric_index_type root_block_size = 2;
    _local_size = root_block_size + static_cast<numeric_index_type>(_comm->rank());
    _global_size = 0;
    _i.push_back(0);
 
    for (processor_id_type p = 0; p < _comm->size(); ++p)
    {
      numeric_index_type block_size = root_block_size + static_cast<numeric_index_type>(p);
      _global_size += block_size;
      _i.push_back(_global_size);
    }
 
    _matrix->init(_global_size,
                  _global_size,
                  _local_size,
                  _local_size,
                  /*nnz=*/_local_size,
                  /*noz=*/0);
  }

References TestCommWorld.

tearDown()

void PetscMatrixTest::tearDown ( )

inline

Definition at line 54 of file petsc_matrix_test.C.

{}

testGetAndSet()

void PetscMatrixTest::testGetAndSet ( )

inline

Definition at line 56 of file petsc_matrix_test.C.

  {
    std::vector<numeric_index_type> rows(_local_size);
    std::vector<numeric_index_type> cols(_local_size);
    DenseMatrix<Number> local(_local_size, _local_size);
 
    numeric_index_type index = _i[_comm->rank()], count = 0;
    for (; count < _local_size; ++count, ++index)
    {
      rows[count] = index;
      cols[count] = index;
      for (numeric_index_type j = 0; j < _local_size; ++j)
        local(count, j) = (count + 1) * (j + 1) * (_comm->rank() + 1);
    }
 
    _matrix->add_matrix(local, rows, cols);
    _matrix->close();
 
    index = _i[_comm->rank()], count = 0;
 
    auto functor = [this]()
    {
      std::vector<numeric_index_type> cols_to_get;
      std::vector<Number> values;
      numeric_index_type local_index = _i[_comm->rank()], local_count = 0;
      for (; local_count < _local_size; ++local_count, ++local_index)
      {
        _matrix->get_row(local_index, cols_to_get, values);
        for (numeric_index_type j = 0; j < _local_size; ++j)
        {
          LIBMESH_ASSERT_FP_EQUAL(libMesh::libmesh_real(values[j]),
                                  (local_count + 1) * (j + 1) * (_comm->rank() + 1),
                                  _tolerance);
          CPPUNIT_ASSERT_EQUAL(cols_to_get[local_count], local_index);
        }
      }
    };
 
#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
  }

References libMesh::libmesh_real().

Member Data Documentation

_comm

Parallel::Communicator* PetscMatrixTest::_comm

private

Definition at line 111 of file petsc_matrix_test.C.

_global_size

numeric_index_type PetscMatrixTest::_global_size

private

Definition at line 113 of file petsc_matrix_test.C.

_i

std::vector<numeric_index_type> PetscMatrixTest::_i

private

Definition at line 114 of file petsc_matrix_test.C.

_local_size

numeric_index_type PetscMatrixTest::_local_size

private

Definition at line 113 of file petsc_matrix_test.C.

_matrix

std::unique_ptr<PetscMatrix<Number> > PetscMatrixTest::_matrix

private

Definition at line 112 of file petsc_matrix_test.C.

_tolerance

const Real PetscMatrixTest::_tolerance = TOLERANCE * TOLERANCE

private

Definition at line 115 of file petsc_matrix_test.C.

---

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

• tests/numerics/petsc_matrix_test.C