EstimatorManagerNewTest Class Reference

Testing class breaking EstimatorManagerNew encapsultation. More...

Collaboration diagram for EstimatorManagerNewTest:

Public Types
using	QMCT = QMCTraits

Public Member Functions
	EstimatorManagerNewTest (const QMCHamiltonian &ham, Communicate *comm, int ranks)
void	fakeMainScalarSamples ()
	Quickly add main scalar samples using FakeEstimator mock estimator. More...
void	fakeScalarSamplesAndCollect ()
	Quickly add scalar samples using FakeEstimator mock estimator. More...
void	fakeSomeOperatorEstimatorSamples (int rank)
	Quickly add scalar samples using FakeOperatorEstimator mock estimator. More...
void	collectMainEstimators ()
	call private EMB method and collect EMBTs estimators_ as main_estimators More...
void	collectScalarEstimators ()
	call private EMB method and colelct EMBTs estimators_ More...
void	collectOperatorEstimators ()
	reduce the OperatorEstimators onto the EstimatorManagerNew copy. More...
std::vector< QMCT::RealType >	generateGoodOperatorData (int num_ranks)
	for mpi test (it's trivial for 1 rank) More...
bool	testReplaceMainEstimator ()
	test replacing the main estimator More...
bool	testMakeBlockAverages ()
void	testReduceOperatorEstimators ()
std::vector< QMCT::RealType > &	get_operator_data ()

Public Attributes
EstimatorManagerNew	em

Private Attributes
Communicate *	comm_
std::vector< FakeEstimator >	estimators_
std::vector< RefVector< ScalarEstimatorBase > >	scalar_estimators_

Detailed Description

Testing class breaking EstimatorManagerNew encapsultation.

Wraps EstimatorManagerNew

Definition at line 30 of file EstimatorManagerNewTest.h.

Member Typedef Documentation

QMCT

using QMCT = QMCTraits

Definition at line 33 of file EstimatorManagerNewTest.h.

Constructor & Destructor Documentation

EstimatorManagerNewTest()

EstimatorManagerNewTest	(	const QMCHamiltonian &	ham,
		Communicate *	comm,
		int	ranks
	)

Definition at line 24 of file EstimatorManagerNewTest.cpp.

References qmcplusplus::app_log(), EstimatorManagerNewTest::comm_, qmcplusplus::num_ranks, and Communicate::size().

    : em(ham, comm), comm_(comm)
{
  int num_ranks = comm_->size();
  if (num_ranks != ranks)
    throw std::runtime_error("Bad Rank Count, test expects different number of ranks.");

  app_log() << "running on " << num_ranks << '\n';
}

Member Function Documentation

collectMainEstimators()

void collectMainEstimators

(

)

call private EMB method and collect EMBTs estimators_ as main_estimators

Definition at line 147 of file EstimatorManagerNewTest.cpp.

References EstimatorManagerNew::collectMainEstimators(), EstimatorManagerNewTest::em, and EstimatorManagerNewTest::estimators_.

Referenced by qmcplusplus::TEST_CASE().

{
  auto crowd_main_estimators = makeRefVector<ScalarEstimatorBase>(estimators_);
  em.collectMainEstimators(crowd_main_estimators);
}

collectOperatorEstimators()

void collectOperatorEstimators

(

)

reduce the OperatorEstimators onto the EstimatorManagerNew copy.

collectScalarEstimators()

void collectScalarEstimators

(

)

call private EMB method and colelct EMBTs estimators_

fakeMainScalarSamples()

void fakeMainScalarSamples

(

)

Quickly add main scalar samples using FakeEstimator mock estimator.

Definition at line 52 of file EstimatorManagerNewTest.cpp.

References EstimatorManagerNewTest::em, EstimatorManagerNewTest::estimators_, EstimatorManagerNew::get_AverageCache(), ScalarEstimatorBase::scalars, and ScalarEstimatorBase::scalars_saved.

Referenced by qmcplusplus::TEST_CASE().

{
  estimators_.clear();
  FakeEstimator fake_estimator;
  fake_estimator.scalars.resize(4);
  fake_estimator.scalars_saved.resize(4);
  fake_estimator.scalars[0](1.0);
  fake_estimator.scalars[1](2.0);
  fake_estimator.scalars[2](3.0);
  fake_estimator.scalars[3](4.0);

  //three estimators
  estimators_.push_back(fake_estimator);
  estimators_.push_back(fake_estimator);
  estimators_.push_back(fake_estimator);

  estimators_[2].scalars[0](2.0);
  estimators_[2].scalars[1](2.0);
  estimators_[2].scalars[2](2.0);
  estimators_[2].scalars[3](2.0);

  em.get_AverageCache().resize(4);
}

fakeScalarSamplesAndCollect()

void fakeScalarSamplesAndCollect

(

)

Quickly add scalar samples using FakeEstimator mock estimator.

Definition at line 76 of file EstimatorManagerNewTest.cpp.

References EstimatorManagerNew::addScalarEstimator(), EstimatorManagerNew::collectScalarEstimators(), EstimatorManagerNewTest::em, EstimatorManagerNewTest::estimators_, EstimatorManagerNew::get_AverageCache(), EstimatorManagerNewTest::scalar_estimators_, ScalarEstimatorBase::scalars, and ScalarEstimatorBase::scalars_saved.

Referenced by qmcplusplus::TEST_CASE().

{
  estimators_.clear();
  FakeEstimator fake_estimator;
  fake_estimator.scalars.resize(4);
  fake_estimator.scalars_saved.resize(4);
  fake_estimator.scalars[0](1.0);
  fake_estimator.scalars[1](2.0);
  fake_estimator.scalars[2](3.0);
  fake_estimator.scalars[3](4.0);

  //three estimators
  estimators_.push_back(fake_estimator);
  estimators_.push_back(fake_estimator);
  estimators_.push_back(fake_estimator);

  estimators_[2].scalars[0](2.0);
  estimators_[2].scalars[1](2.0);
  estimators_[2].scalars[2](2.0);
  estimators_[2].scalars[3](2.0);

  std::vector<FakeEstimator> estimators2;
  estimators2.push_back(fake_estimator);
  estimators2.push_back(fake_estimator);
  estimators2.push_back(fake_estimator);

  std::vector<FakeEstimator> estimators3;
  estimators3.push_back(fake_estimator);
  estimators3.push_back(fake_estimator);
  estimators3.push_back(fake_estimator);

  em.addScalarEstimator(std::make_unique<FakeEstimator>());
  em.addScalarEstimator(std::make_unique<FakeEstimator>());
  em.addScalarEstimator(std::make_unique<FakeEstimator>());

  scalar_estimators_.push_back(makeRefVector<ScalarEstimatorBase>(estimators_));
  scalar_estimators_.push_back(makeRefVector<ScalarEstimatorBase>(estimators2));
  scalar_estimators_.push_back(makeRefVector<ScalarEstimatorBase>(estimators3));

  em.get_AverageCache().resize(4);

  em.collectScalarEstimators(scalar_estimators_);
}

fakeSomeOperatorEstimatorSamples()

void fakeSomeOperatorEstimatorSamples

(

int

rank

)

Quickly add scalar samples using FakeOperatorEstimator mock estimator.

Definition at line 120 of file EstimatorManagerNewTest.cpp.

References EstimatorManagerNewTest::comm_, qmcplusplus::crowd, EstimatorManagerNewTest::em, OperatorEstBase::get_data(), EstimatorManagerNew::operator_ests_, qmcplusplus::rank, FakeOperatorEstimator::set_walker_weights(), and Communicate::size().

{
  em.operator_ests_.emplace_back(new FakeOperatorEstimator(comm_->size(), DataLocality::crowd));
  FakeOperatorEstimator& foe        = dynamic_cast<FakeOperatorEstimator&>(*(em.operator_ests_.back()));
  std::vector<QMCT::RealType>& data = foe.get_data();
  for (int id = 0; id < data.size(); ++id)
  {
    if (id > rank)
      data[id] += rank + 1;
  }
  foe.set_walker_weights(1);
}

generateGoodOperatorData()

std::vector< QMCTraits::RealType > generateGoodOperatorData

(

int

num_ranks

)

for mpi test (it's trivial for 1 rank)

only used by test_manager_mpi.cpp so implemented there.

Definition at line 133 of file EstimatorManagerNewTest.cpp.

References qmcplusplus::num_ranks.

{
  std::vector<QMCT::RealType> data(num_ranks * 10, 0.0);
  for (int ir = 0; ir < num_ranks; ++ir)
  {
    for (int id = 0; id < data.size(); ++id)
    {
      if (id > ir)
        data[id] += ir + 1;
    }
  }
  return data;
}

get_operator_data()

std::vector<QMCT::RealType>& get_operator_data ( )

inline

Definition at line 59 of file EstimatorManagerNewTest.h.

References EstimatorManagerNewTest::em, and EstimatorManagerNew::operator_ests_.

{ return em.operator_ests_[0]->get_data(); }

testMakeBlockAverages()

bool testMakeBlockAverages

(

)

Definition at line 24 of file test_manager_mpi.cpp.

References EstimatorManagerNew::acceptRatioInd, RecordNamedProperty< T >::add(), EstimatorManagerNew::BlockProperties, EstimatorManagerNew::collectMainEstimators(), EstimatorManagerNew::cpuInd, EstimatorManagerNewTest::em, EstimatorManagerNewTest::estimators_, EstimatorManagerNew::makeBlockAverages(), EstimatorManagerNew::my_comm_, EstimatorManagerNew::PropertyCache, Communicate::rank(), Vector< T, Alloc >::resize(), RecordNamedProperty< T >::size(), and EstimatorManagerNew::weightInd.

{
  if (em.my_comm_->rank() == 1)
  {
    estimators_[1].scalars[0](3.0);
    estimators_[1].scalars[1](3.0);
    estimators_[1].scalars[2](3.0);
    estimators_[1].scalars[3](3.0);
  }

  // manipulation of state to arrive at to be tested state.
  // - From EstimatorManagerBase::reset
  em.weightInd      = em.BlockProperties.add("BlockWeight");
  em.cpuInd         = em.BlockProperties.add("BlockCPU");
  em.acceptRatioInd = em.BlockProperties.add("AcceptRatio");

  // - From EstimatorManagerBase::start
  em.PropertyCache.resize(em.BlockProperties.size());

  // - From EstimatorManagerBase::stopBlocknew
  //   three estimators
  //   - 2 with 1 sample 1
  //   - 1 with 2
  double block_weight = 0;
  std::for_each(estimators_.begin(), estimators_.end(),
                [&block_weight](auto& est) { block_weight += est.scalars[0].count(); });
  em.PropertyCache[em.weightInd] = block_weight;
  em.PropertyCache[em.cpuInd]    = 1.0;

  //RefVector<ScalarEstimatorBase> est_list = makeRefVector<ScalarEstimatorBase>(estimators_);
  RefVector<ScalarEstimatorBase> est_lists(makeRefVector<ScalarEstimatorBase>(estimators_));
  em.collectMainEstimators(est_lists);

  unsigned long accepts = 4;
  unsigned long rejects = 1;
  em.makeBlockAverages(accepts, rejects);
  return true;
}

testReduceOperatorEstimators()

void testReduceOperatorEstimators

(

)

Definition at line 153 of file EstimatorManagerNewTest.cpp.

References EstimatorManagerNewTest::em, and EstimatorManagerNew::reduceOperatorEstimators().

{ em.reduceOperatorEstimators(); }

testReplaceMainEstimator()

bool testReplaceMainEstimator

(

)

test replacing the main estimator

Definition at line 34 of file EstimatorManagerNewTest.cpp.

References EstimatorManagerNew::addMainEstimator(), qmcplusplus::CHECK(), EstimatorManagerNewTest::em, and EstimatorManagerNew::main_estimator_.

Referenced by qmcplusplus::TEST_CASE().

{
  // Must create on heap since the EstimatorManager destructor deletes all estimators
  auto fake_est_uptr      = std::make_unique<FakeEstimator>();
  FakeEstimator* fake_est = fake_est_uptr.get();

  em.addMainEstimator(std::move(fake_est_uptr));

  CHECK(em.main_estimator_->getSubTypeStr() == "fake");

  auto fake_est2   = std::make_unique<FakeEstimator>();
  fake_est2->type_ = "Fake2";

  em.addMainEstimator(std::move(fake_est2));

  return em.main_estimator_->getSubTypeStr() == "Fake2";
}

Member Data Documentation

comm_

Communicate* comm_

private

Definition at line 63 of file EstimatorManagerNewTest.h.

Referenced by EstimatorManagerNewTest::EstimatorManagerNewTest(), and EstimatorManagerNewTest::fakeSomeOperatorEstimatorSamples().

em

EstimatorManagerNew em

Definition at line 61 of file EstimatorManagerNewTest.h.

Referenced by EstimatorManagerNewTest::collectMainEstimators(), EstimatorManagerNewTest::fakeMainScalarSamples(), EstimatorManagerNewTest::fakeScalarSamplesAndCollect(), EstimatorManagerNewTest::fakeSomeOperatorEstimatorSamples(), EstimatorManagerNewTest::get_operator_data(), qmcplusplus::if(), qmcplusplus::TEST_CASE(), EstimatorManagerNewTest::testMakeBlockAverages(), EstimatorManagerNewTest::testReduceOperatorEstimators(), and EstimatorManagerNewTest::testReplaceMainEstimator().

estimators_

std::vector<FakeEstimator> estimators_

private

Definition at line 64 of file EstimatorManagerNewTest.h.

Referenced by EstimatorManagerNewTest::collectMainEstimators(), EstimatorManagerNewTest::fakeMainScalarSamples(), EstimatorManagerNewTest::fakeScalarSamplesAndCollect(), and EstimatorManagerNewTest::testMakeBlockAverages().

scalar_estimators_

std::vector<RefVector<ScalarEstimatorBase> > scalar_estimators_

private

Definition at line 65 of file EstimatorManagerNewTest.h.

Referenced by EstimatorManagerNewTest::fakeScalarSamplesAndCollect().

---

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

• /home/pk7/projects/qmc/for_cron_doxygen/qmcpack/src/Estimators/tests/EstimatorManagerNewTest.h
• /home/pk7/projects/qmc/for_cron_doxygen/qmcpack/src/Estimators/tests/EstimatorManagerNewTest.cpp
• /home/pk7/projects/qmc/for_cron_doxygen/qmcpack/src/Estimators/tests/test_manager_mpi.cpp