SelfAwareBranch.cc File Reference

#include <boost/test/included/unit_test.hpp>
#include <boost/exception/all.hpp>
#include <filesystem>
#include "Flow.h"
#include "Jet.h"
#include "SelfAwareBranch.h"

Include dependency graph for SelfAwareBranch.cc:

Macros
#define	BOOST_TEST_MODULE   SelfAwareBranch

Functions
	BOOST_AUTO_TEST_CASE (plain_type)
	BOOST_AUTO_TEST_CASE (vector_type)
	BOOST_AUTO_TEST_CASE (polymorphism)

Macro Definition Documentation

BOOST_TEST_MODULE

#define BOOST_TEST_MODULE   SelfAwareBranch

Function Documentation

BOOST_AUTO_TEST_CASE() [1/3]

BOOST_AUTO_TEST_CASE

(

plain_type

)

    {
        BOOST_TEST_MESSAGE( "Filling the tree" );
        {
            Flow flow(verbose);
            auto tOut = flow.GetOutputTree("self_aware_branch.root");
            SelfAwarePlainBranch<int, int> br1(flow, "branch1", [](int v) -> int { return v;});
            br1.Fill(42);
            SelfAwarePlainBranch<int> br2(flow, "branch2", []() -> int { return 43;});
            br2.Fill();
            tOut->Fill();
            br1.Clear();
            br1.Fill(43);
            tOut->Fill();
        }
 
        BOOST_TEST_MESSAGE( "Testing output values" );
        {
            Flow flow(verbose, {"self_aware_branch.root"});
            auto tIn = flow.GetInputTree();
            auto br1 = flow.GetBranchReadOnly<int>("branch1"),
                 br2 = flow.GetBranchReadOnly<int>("branch2");
            tIn->Scan();
            tIn->GetEntry(0);
            BOOST_TEST( *br1 == 42 );
            BOOST_TEST( *br2 == 43 );
            tIn->GetEntry(1);
            BOOST_TEST( *br1 == 43 );
            BOOST_TEST( *br2 == 43 );
        }
 
        fs::remove("self_aware_branch.root");
    }

BOOST_AUTO_TEST_CASE() [2/3]

BOOST_AUTO_TEST_CASE

(

polymorphism

)

    {
        BOOST_TEST_MESSAGE( "Filling the tree" );
        {
            Flow flow(verbose);
            auto tOut = flow.GetOutputTree("self_aware_branch.root");
            auto fill = [](int v) -> int { return v;};
 
            SelfAwareBranch<int> * plain_branch  = new SelfAwarePlainBranch <int, int>(flow, "plain_branch" , fill);
            SelfAwareBranch<int> * vector_branch = new SelfAwareVectorBranch<int, int>(flow, "vector_branch", fill);
            for (auto&& branch: {plain_branch, vector_branch})
                branch->Fill(42);
            tOut->Fill();
            for (auto&& branch: {plain_branch, vector_branch})
                delete branch;
        }
 
        BOOST_TEST_MESSAGE( "Testing output values" );
        {
            Flow flow(verbose, {"self_aware_branch.root"});
            auto tIn = flow.GetInputTree();
            auto plain_branch  = flow.GetBranchReadOnly<int>        ("plain_branch");
            auto vector_branch = flow.GetBranchReadOnly<vector<int>>("vector_branch");
            tIn->GetEntry(0);
            tIn->Scan();
            BOOST_TEST( *plain_branch        == 42 );
            BOOST_TEST( vector_branch->at(0) == 42 );
        }
 
        fs::remove("self_aware_branch.root");
    }

BOOST_AUTO_TEST_CASE() [3/3]

BOOST_AUTO_TEST_CASE

(

vector_type

)

    {
        BOOST_TEST_MESSAGE( "Filling the tree" );
        {
            Flow flow(verbose);
            auto tOut = flow.GetOutputTree("self_aware_branch.root");
            SelfAwareVectorBranch<int, int> br1(flow, "branch1", [](int v) -> int { return v;});
            br1.Fill(42);
            br1.Fill(44);
            SelfAwareVectorBranch<int> br2(flow, "branch2", []() -> int { return 43;});
            br2.Fill();
            br2.Fill();
            tOut->Fill();
            br1.Clear();
            br2.Clear();
            br1.Fill(24);
            tOut->Fill();
        }
 
        BOOST_TEST_MESSAGE( "Testing output values" );
        {
            Flow flow(verbose, {"self_aware_branch.root"});
            auto tIn = flow.GetInputTree();
            auto br1 = flow.GetBranchReadOnly<vector<int>>("branch1"),
                 br2 = flow.GetBranchReadOnly<vector<int>>("branch2");
            tIn->Scan();
            tIn->GetEntry(0);
            BOOST_TEST( br1->at(0) == 42 );
            BOOST_TEST( br2->at(0) == 43 );
            BOOST_TEST( br1->at(1) == 44 );
            BOOST_TEST( br2->at(1) == 43 );
            tIn->GetEntry(1);
            BOOST_TEST( br1->at(0) == 24 );
            BOOST_TEST( br2->size() == 0 );
        }
 
        fs::remove("self_aware_branch.root");
    }