Tools/test/TestJetReader.h

#include "cute/cute.h"
#include "cute/cute_suite.h"

#include <boost/scoped_ptr.hpp>
#include <boost/shared_ptr.hpp>

#include "../interface/Readers/JetReader.h"
#include "InputFiles.h"
using namespace BAT;

struct TestJetReader {
private:
        boost::shared_ptr<TChain> input;
        boost::scoped_ptr<JetReader> reader, PFreader;
        JetCollection jets, pfjets;
        JetPointer firstJet, firstPFJet;

public:
        TestJetReader() :
                input(new TChain(NTupleEventReader::EVENT_CHAIN)),
                reader(new JetReader(input)),
                PFreader(new JetReader(input, JetAlgorithm::CA08PF)),
                jets(),
                pfjets(),
                firstJet(),
                firstPFJet() {
                input->Add(InputFile::ttbar);
                input->SetBranchStatus("*", 0);
                reader->initialise();
                PFreader->initialise();
                input->GetEntry(1);
                jets = reader->getJets();
                pfjets = PFreader->getJets();
                firstJet = jets.front();
                firstPFJet = pfjets.front();
        }

        void testReadJetsSize() {
                ASSERT_EQUAL(10, jets.size());
        }

        void testReadFirstJetEnergy() {
                ASSERT_EQUAL_DELTA(138.88, firstJet->energy(), 0.001);
        }

        void testReadFirstJetEMF() {
                ASSERT_EQUAL_DELTA(0.568266, firstJet->emf(), 0.00001);
        }

        void testReadFirstJetn90Hits() {
                ASSERT_EQUAL_DELTA(37, firstJet->n90Hits(), 0.1);
        }

        void testReadFirstJetfHPD() {
                ASSERT_EQUAL_DELTA(0.262745, firstJet->fHPD(), 0.00001);
        }

        void testReadPFJetsSize() {
                ASSERT_EQUAL(12, pfjets.size());
        }

        void testReadFirstPFJetEnergy() {
                ASSERT_EQUAL_DELTA(139.809, firstPFJet->energy(), 0.001);
        }

        void testReadFirstPFJetEMF() {
                ASSERT_EQUAL_DELTA(0, firstPFJet->emf(), 0.00001);
        }

        void testReadFirstPFJetn90Hits() {
                ASSERT_EQUAL_DELTA(0, firstPFJet->n90Hits(), 0.1);
        }

        void testReadFirstPFJetfHPD() {
                ASSERT_EQUAL_DELTA(0, firstPFJet->fHPD(), 0.00001);
        }

        void testUsedAlgorithmDefault(){
                ASSERT_EQUAL(JetAlgorithm::Calo_AntiKT_Cone05, firstJet->getUsedAlgorithm());
        }

        void testUsedAlgorithmPFJet(){
                ASSERT_EQUAL(JetAlgorithm::CA08PF, firstPFJet->getUsedAlgorithm());
        }

};
extern cute::suite make_suite_TestJetReader() {
        cute::suite s;

        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadJetsSize));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetEnergy));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetEMF));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetn90Hits));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetfHPD));

        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadPFJetsSize));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetEnergy));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetEMF));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetn90Hits));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetfHPD));

        s.push_back(CUTE_SMEMFUN(TestJetReader, testUsedAlgorithmDefault));
        s.push_back(CUTE_SMEMFUN(TestJetReader, testUsedAlgorithmPFJet));

        return s;
}
Revision:	1.1
Committed:	Thu Dec 1 16:28:47 2011 UTC (13 years, 5 months ago) by dhidas
Content type:	text/plain
Branch point for:	dhidas, MAIN
Log Message:	Initial revision
#	User	Rev	Content
1	dhidas	1.1	#include "cute/cute.h"
2			#include "cute/cute_suite.h"
3
4			#include <boost/scoped_ptr.hpp>
5			#include <boost/shared_ptr.hpp>
6
7			#include "../interface/Readers/JetReader.h"
8			#include "InputFiles.h"
9			using namespace BAT;
10
11			struct TestJetReader {
12			private:
13			boost::shared_ptr<TChain> input;
14			boost::scoped_ptr<JetReader> reader, PFreader;
15			JetCollection jets, pfjets;
16			JetPointer firstJet, firstPFJet;
17
18			public:
19			TestJetReader() :
20			input(new TChain(NTupleEventReader::EVENT_CHAIN)),
21			reader(new JetReader(input)),
22			PFreader(new JetReader(input, JetAlgorithm::CA08PF)),
23			jets(),
24			pfjets(),
25			firstJet(),
26			firstPFJet() {
27			input->Add(InputFile::ttbar);
28			input->SetBranchStatus("*", 0);
29			reader->initialise();
30			PFreader->initialise();
31			input->GetEntry(1);
32			jets = reader->getJets();
33			pfjets = PFreader->getJets();
34			firstJet = jets.front();
35			firstPFJet = pfjets.front();
36			}
37
38			void testReadJetsSize() {
39			ASSERT_EQUAL(10, jets.size());
40			}
41
42			void testReadFirstJetEnergy() {
43			ASSERT_EQUAL_DELTA(138.88, firstJet->energy(), 0.001);
44			}
45
46			void testReadFirstJetEMF() {
47			ASSERT_EQUAL_DELTA(0.568266, firstJet->emf(), 0.00001);
48			}
49
50			void testReadFirstJetn90Hits() {
51			ASSERT_EQUAL_DELTA(37, firstJet->n90Hits(), 0.1);
52			}
53
54			void testReadFirstJetfHPD() {
55			ASSERT_EQUAL_DELTA(0.262745, firstJet->fHPD(), 0.00001);
56			}
57
58			void testReadPFJetsSize() {
59			ASSERT_EQUAL(12, pfjets.size());
60			}
61
62			void testReadFirstPFJetEnergy() {
63			ASSERT_EQUAL_DELTA(139.809, firstPFJet->energy(), 0.001);
64			}
65
66			void testReadFirstPFJetEMF() {
67			ASSERT_EQUAL_DELTA(0, firstPFJet->emf(), 0.00001);
68			}
69
70			void testReadFirstPFJetn90Hits() {
71			ASSERT_EQUAL_DELTA(0, firstPFJet->n90Hits(), 0.1);
72			}
73
74			void testReadFirstPFJetfHPD() {
75			ASSERT_EQUAL_DELTA(0, firstPFJet->fHPD(), 0.00001);
76			}
77
78			void testUsedAlgorithmDefault(){
79			ASSERT_EQUAL(JetAlgorithm::Calo_AntiKT_Cone05, firstJet->getUsedAlgorithm());
80			}
81
82			void testUsedAlgorithmPFJet(){
83			ASSERT_EQUAL(JetAlgorithm::CA08PF, firstPFJet->getUsedAlgorithm());
84			}
85
86			};
87			extern cute::suite make_suite_TestJetReader() {
88			cute::suite s;
89
90			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadJetsSize));
91			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetEnergy));
92			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetEMF));
93			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetn90Hits));
94			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstJetfHPD));
95
96			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadPFJetsSize));
97			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetEnergy));
98			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetEMF));
99			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetn90Hits));
100			s.push_back(CUTE_SMEMFUN(TestJetReader, testReadFirstPFJetfHPD));
101
102			s.push_back(CUTE_SMEMFUN(TestJetReader, testUsedAlgorithmDefault));
103			s.push_back(CUTE_SMEMFUN(TestJetReader, testUsedAlgorithmPFJet));
104
105			return s;
106			}