EL::UnitTestFixture Class Reference

#include <UnitTestFixture.h>

Inheritance diagram for EL::UnitTestFixture:

Collaboration diagram for EL::UnitTestFixture:

Public Member Functions
std::string	getJob ()
std::shared_ptr< SH::Sample >	getSample (const std::string &sameName)
SH::SampleHandler	getSH ()
unsigned	eventCount (const std::string &sampleName)
TObject *	getTObject (const std::string &sampleName, const std::string &objectName, bool isMandatory)
template<typename T>
T *	getHist (const std::string &sampleName, const std::string &objectName, bool isMandatory)
TH1 *	getCallbacks (const std::string &sampleName)
void	checkFileExecuted (const std::string &sampleName)

Private Member Functions
std::string	makeFile (const std::vector< unsigned > &entries)

Static Private Attributes
static std::map< std::shared_ptr< Driver >, std::string >	m_jobs

Detailed Description

Definition at line 22 of file UnitTestFixture.h.

Member Function Documentation

checkFileExecuted()

void EL::UnitTestFixture::checkFileExecuted

(

const std::string &

sampleName

)

Definition at line 210 of file UnitTestFixture.cxx.

  {
    std::set<std::string> filesOut;
    {
      SH::SampleHandler sh;
      sh.load (getJob() + "/hist");
      SH::Sample *sample = sh.get (sampleName);
      if (sample == nullptr)
    RCU_THROW_MSG ("couldn't find sample: " + sampleName);
      auto vec = readVectorFromTree (sample, "EventLoop_FileExecuted", "file");
      filesOut.insert (vec.begin(), vec.end());
    }
    std::set<std::string> filesIn;
    for (auto& file : getSample (sampleName)->makeFileList())
    {
      auto split = file.rfind ('/');
      if (split == std::string::npos)
    split = 0;
      else
    ++ split;
      std::string fileName = file.substr (split);
      ASSERT_TRUE (filesIn.find (fileName) == filesIn.end());
      filesIn.insert (fileName);
    }
    ASSERT_EQ (filesIn, filesOut);
  }

eventCount()

unsigned EL::UnitTestFixture::eventCount

(

const std::string &

sampleName

)

Definition at line 193 of file UnitTestFixture.cxx.

  {
    TH1 *hist = getHist<TH1> (sampleName, "EventLoop_EventCount", true);
    return hist->GetBinContent (1);
  }

getCallbacks()

TH1 * EL::UnitTestFixture::getCallbacks

(

const std::string &

sampleName

)

Definition at line 202 of file UnitTestFixture.cxx.

  {
    return getHist<TH1> (sampleName, "callbacks", true);
  }

getHist()

template<typename T>

T * EL::UnitTestFixture::getHist	(	const std::string &	sampleName,
		const std::string &	objectName,
		bool	isMandatory )

getJob()

std::string EL::UnitTestFixture::getJob

(

)

Definition at line 138 of file UnitTestFixture.cxx.

  {
    using namespace asg::msgUserCode;
    ANA_MSG_INFO ("in EventLoopTest");
 
    std::shared_ptr<Driver> driver = GetParam().m_driver;
    auto iter = m_jobs.find (driver);
    if (iter != m_jobs.end())
      return iter->second;
 
    Job job;
    job.sampleHandler (getSH());
    GetParam().setupJob (job);
    {
      EL::AnaAlgorithmConfig config;
      config.setType ("EL::UnitTestAlg2");
      config.setName ("newAlg");
      config.setUseXAODs (false);
      ANA_CHECK_THROW (config.setProperty ("property", 42));
      job.algsAdd (config);
    }
    {
      std::unique_ptr<UnitTestAlg1> alg (new UnitTestAlg1);
      job.algsAdd (alg.release());
    }
 
    std::ostringstream submit;
    submit << "submit-" << driver.get();
    driver->submit (job, submit.str());
    driver->wait (submit.str());
    m_jobs[driver] = submit.str();
    return submit.str();
  }

getSample()

std::shared_ptr< SH::Sample > EL::UnitTestFixture::getSample

(

const std::string &

sameName

)

Definition at line 75 of file UnitTestFixture.cxx.

  {
    if (sampleName == "empty")
    {
      static std::shared_ptr<SH::Sample> result;
      static std::once_flag flag;
      std::call_once (flag, [this]() {
        auto myresult = std::make_shared<SH::SampleLocal> ("empty");
        myresult->add (makeFile ({}));
        result = std::move (myresult);
      });
      return result;
    }
    if (sampleName == "single")
    {
      static std::shared_ptr<SH::Sample> result;
      static std::once_flag flag;
      std::call_once (flag, [this]() {
        auto myresult = std::make_shared<SH::SampleLocal> ("single");
        std::vector<unsigned> entries;
        for (unsigned iter = 0; iter != 10000; ++ iter)
          entries.push_back (iter % 10);
        myresult->add (makeFile (entries));
        result = std::move (myresult);
      });
      return result;
    }
    if (sampleName == "multi")
    {
      static std::shared_ptr<SH::Sample> result;
      static std::once_flag flag;
      std::call_once (flag, [this]() {
        auto myresult = std::make_shared<SH::SampleLocal> ("multi");
        for (unsigned jter = 0; jter != 10; ++ jter)
        {
          std::vector<unsigned> entries;
          for (unsigned iter = 0; iter != 10000; ++ iter)
            entries.push_back (iter % 10);
          myresult->add (makeFile (entries));
        }
        result = std::move (myresult);
      });
      return result;
    }
    RCU_THROW_MSG ("unknown sample: " + sampleName);
  }

getSH()

SH::SampleHandler EL::UnitTestFixture::getSH

(

)

Definition at line 125 of file UnitTestFixture.cxx.

  {
    SH::SampleHandler sh;
    sh.add (getSample ("empty"));
    sh.add (getSample ("single"));
    sh.add (getSample ("multi"));
    sh.setMetaString (SH::MetaFields::treeName, "physics");
    return sh;
  }

getTObject()

TObject * EL::UnitTestFixture::getTObject	(	const std::string &	sampleName,
		const std::string &	objectName,
		bool	isMandatory )

Definition at line 175 of file UnitTestFixture.cxx.

  {
    SH::SampleHandler sh;
    sh.load (getJob() + "/hist");
    SH::Sample *sample = sh.get (sampleName);
    if (sample == nullptr)
      RCU_THROW_MSG ("couldn't find sample: " + sampleName);
    TObject *object = sample->readHist (objectName);
    if (isMandatory && object == nullptr)
      RCU_THROW_MSG ("couldn't find object: " + objectName);
    return object;
  }

makeFile()

std::string EL::UnitTestFixture::makeFile ( const std::vector< unsigned > & entries )

private

Definition at line 240 of file UnitTestFixture.cxx.

  {
    static unsigned index = 0;
    std::ostringstream fileName;
    fileName << "file-" << ++ index << ".root";
 
    std::unique_ptr<SH::DiskWriter> file
      = GetParam().make_file_writer (fileName.str());
    {
      if (!entries.empty())
      {
    TTree *tree = new TTree ("physics", "physics");
    Int_t el_n = 0;
    tree->Branch ("el_n", &el_n, "el_n/I");
    for (auto entry : entries)
    {
      el_n = entry;
      tree->Fill ();
    }
      }
      file->file()->Write ();
      file->close ();
    }
    return file->path();
  }

Member Data Documentation

m_jobs

std::map< std::shared_ptr< Driver >, std::string > EL::UnitTestFixture::m_jobs

staticprivate

Definition at line 47 of file UnitTestFixture.h.

---

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

• UnitTestFixture.h
• UnitTestFixture.cxx