1 |
dhidas |
1.1 |
#include "cute/cute.h"
|
2 |
|
|
#include "cute/cute_suite.h"
|
3 |
|
|
|
4 |
|
|
#include "../interface/Readers/METReader.h"
|
5 |
|
|
#include "../interface/RecoObjects/MET.h"
|
6 |
|
|
#include "InputFiles.h"
|
7 |
|
|
using namespace BAT;
|
8 |
|
|
|
9 |
|
|
struct TestMETReader {
|
10 |
|
|
private:
|
11 |
|
|
boost::shared_ptr<TChain> input;
|
12 |
|
|
boost::scoped_ptr<METReader> reader, PFreader;
|
13 |
|
|
METPointer met, pfmet;
|
14 |
|
|
|
15 |
|
|
public:
|
16 |
|
|
TestMETReader() :
|
17 |
|
|
input(new TChain(NTupleEventReader::EVENT_CHAIN)),
|
18 |
|
|
reader(new METReader(input)),
|
19 |
|
|
PFreader(new METReader(input, METAlgorithm::ParticleFlowMET)),
|
20 |
|
|
met(),
|
21 |
|
|
pfmet() {
|
22 |
|
|
input->Add(InputFile::ttbar);
|
23 |
|
|
input->SetBranchStatus("*", 0);
|
24 |
|
|
reader->initialise();
|
25 |
|
|
PFreader->initialise();
|
26 |
|
|
input->GetEntry(0);
|
27 |
|
|
met = reader->getMET();
|
28 |
|
|
pfmet = PFreader->getMET();
|
29 |
|
|
}
|
30 |
|
|
|
31 |
|
|
void testMET() {
|
32 |
|
|
ASSERT_EQUAL_DELTA(69.2572, met->et(), 0.0001);
|
33 |
|
|
}
|
34 |
|
|
|
35 |
|
|
void testPFMET() {
|
36 |
|
|
ASSERT_EQUAL_DELTA(86.9846, pfmet->et(), 0.0001);
|
37 |
|
|
}
|
38 |
|
|
|
39 |
|
|
|
40 |
|
|
void testAreMETsGood(){
|
41 |
|
|
ASSERT_EQUAL(true, met->isGood());
|
42 |
|
|
ASSERT_EQUAL(true, pfmet->isGood());
|
43 |
|
|
}
|
44 |
|
|
|
45 |
|
|
};
|
46 |
|
|
|
47 |
|
|
extern cute::suite make_suite_TestMETReader() {
|
48 |
|
|
cute::suite s;
|
49 |
|
|
|
50 |
|
|
s.push_back(CUTE_SMEMFUN(TestMETReader, testMET));
|
51 |
|
|
s.push_back(CUTE_SMEMFUN(TestMETReader, testPFMET));
|
52 |
|
|
s.push_back(CUTE_SMEMFUN(TestMETReader, testAreMETsGood));
|
53 |
|
|
|
54 |
|
|
return s;
|
55 |
|
|
}
|