ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/dhidas/OSUAnalysis/Tools/test/TestTopPairEventCandidate.h
Revision: 1.1.1.1 (vendor branch)
Committed: Thu Dec 1 16:28:47 2011 UTC (13 years, 5 months ago) by dhidas
Content type: text/plain
Branch: dhidas, MAIN
CVS Tags: START, HEAD
Changes since 1.1: +0 -0 lines
Log Message: 
osu copy modified

File Contents

# Content
1 #include "cute/cute.h"
2 #include "cute/cute_suite.h"
3
4 #include <boost/scoped_ptr.hpp>
5
6 #include "../interface/Event.h"
7 #include "../interface/OnlyForTests/DummyTTbarEvent.h"
8 #include "TestObjectFactory.h"
9 #include <iostream>
10
11 using namespace BAT;
12 using namespace std;
13
14 struct TestTopPairEventCandidate {
15 DummyTTbarEvent ttbarEvent, goodZEvent, poorZEvent, DiJetEvent, DiJetEventWithConversion, muonEvent;
16 DummyTTbarEvent emptyEvent, customEvent;
17
18 ElectronPointer goodIsolatedElectron, goodIsolatedElectron2;
19 ElectronPointer goodLooseElectron;
20 ElectronPointer badElectron;
21 ElectronPointer electronFromConversion;
22 JetPointer goodJet, goodBJet, badJet, goodJetCloseToElectron;
23 VertexPointer goodVertex;
24 VertexPointer badVertex;
25 MuonPointer goodIsolatedMuon;
26 MuonPointer badMuon;
27 METPointer met;
28 TrackCollection lessThan10Tracks;
29 TrackCollection moreThan10TracksHighPurity;
30 TrackCollection moreThan10TracksLowPurity;
31 TrackCollection moreThan10TracksMixedPurity_H;
32 TrackCollection moreThan10TracksMixedPurity_L;
33
34 TestTopPairEventCandidate() :
35 ttbarEvent(),
36 goodZEvent(),
37 poorZEvent(),
38 DiJetEvent(),
39 DiJetEventWithConversion(),
40 muonEvent(),
41 emptyEvent(),
42 customEvent(),
43 goodIsolatedElectron(TestObjectFactory::goodIsolatedElectron()),
44 goodIsolatedElectron2(TestObjectFactory::goodIsolatedElectron2()),
45 goodLooseElectron(new Electron(100., 79., -13., -5.)),
46 badElectron(new Electron(20, 14., 15., 0)),
47 electronFromConversion(new Electron(*goodIsolatedElectron)),
48 goodJet(new Jet(100, 13, 99, 5)),
49 goodBJet(new Jet(*goodJet)),
50 badJet(new Jet(20, 19, 0, 0)),
51 goodJetCloseToElectron(new Jet(100., 98., 13., 5.)),
52 goodVertex(TestObjectFactory::goodVertex()),
53 badVertex(TestObjectFactory::badFakeVertex()),
54 goodIsolatedMuon(new Muon(100., 99., 13., 5.)),
55 badMuon(new Muon(100., 99., 13., 5.)),
56 met(new MET(40, 30)) {
57 setUpGoodLooseElectron();
58 setUpBadElectron();
59 setUpGoodIsolatedElectronFromConversion();
60 setUpGoodJet();
61 setUpGoodBJet();
62 setUpGoodJetCloseToElectron();
63 setUpIsolatedGoodMuon();
64 setUpBadMuon();
65 setUpTracks();
66
67 setUpTTbarEvent();
68 setUpGoodZEvent();
69 setUpPoorZEvent();
70 setUpDiJetEvent();
71 setUpMuonEvent();
72 setUpCustomEvent();
73 }
74
75 private:
76
77 void setUpBadElectron() {
78 badElectron->setHcalIsolation(4);
79 badElectron->setEcalIsolation(44);
80 badElectron->setTrackerIsolation(1);
81 badElectron->setZDistanceToPrimaryVertex(0);
82 }
83
84 void setUpGoodLooseElectron() {
85 goodLooseElectron->setHcalIsolation(5);
86 goodLooseElectron->setEcalIsolation(3);
87 goodLooseElectron->setTrackerIsolation(4);
88 goodLooseElectron->setSigmaIEtaIEta(0.009);
89 goodLooseElectron->setDPhiIn(0.7);
90 goodLooseElectron->setDEtaIn(0.006);
91 goodLooseElectron->setHadOverEm(0.14);
92 goodLooseElectron->setRobustLooseID(true);
93 goodLooseElectron->setNumberOfMissingInnerLayerHits(0);
94 goodLooseElectron->setSuperClusterEta(0);
95 goodLooseElectron->setZDistanceToPrimaryVertex(0);
96 TrackPointer track = TrackPointer(new Track(goodLooseElectron->phi(), goodLooseElectron->eta(),
97 goodLooseElectron->pt(), goodLooseElectron->theta()));
98 track->setD0(goodLooseElectron->d0_wrtBeamSpot());
99 track->setCharge(goodLooseElectron->charge());
100 assert(goodLooseElectron->isGood() == false);
101 assert(goodLooseElectron->isLoose() && goodLooseElectron->relativeIsolation() < 1.);
102 }
103
104 void setUpGoodIsolatedElectronFromConversion() {
105 electronFromConversion->setHcalIsolation(0.5);
106 electronFromConversion->setEcalIsolation(0.3);
107 electronFromConversion->setTrackerIsolation(0.4);
108 electronFromConversion->setNumberOfMissingInnerLayerHits(3);
109 electronFromConversion->setZDistanceToPrimaryVertex(0);
110 }
111 void setUpGoodJet() {
112 goodJet->setEMF(0.2);
113 goodJet->setFHPD(0.5);
114 goodJet->setN90Hits(2);
115 }
116
117 void setUpGoodBJet() {
118 goodBJet->setEMF(0.2);
119 goodBJet->setFHPD(0.5);
120 goodBJet->setN90Hits(2);
121 goodBJet->setDiscriminatorForBtagType(2.5, BtagAlgorithm::SimpleSecondaryVertexHighEffBTag);
122 }
123
124 void setUpGoodJetCloseToElectron() {
125 goodJetCloseToElectron->setEMF(0.2);
126 goodJetCloseToElectron->setFHPD(0.5);
127 goodJetCloseToElectron->setN90Hits(2);
128 }
129
130
131 void setUpIsolatedGoodMuon() {
132 goodIsolatedMuon->makeGlobal(true);
133 goodIsolatedMuon->setEcalIsolation(1);
134 goodIsolatedMuon->setHcalIsolation(1);
135 goodIsolatedMuon->setTrackerIsolation(1);
136 }
137
138 void setUpBadMuon() {
139 badMuon->makeGlobal(false);
140 }
141
142 void setUpTTbarEvent() {
143 VertexCollection vertices;
144 vertices.push_back(goodVertex);
145 ttbarEvent.setVertices(vertices);
146 ElectronCollection electrons;
147 electrons.push_back(goodIsolatedElectron);
148 electrons.push_back(badElectron);
149 ttbarEvent.setElectrons(electrons);
150 JetCollection jets;
151 jets.push_back(goodJet);
152 jets.push_back(goodJet);
153 jets.push_back(goodBJet);
154 jets.push_back(goodBJet);
155 jets.push_back(badJet);
156 ttbarEvent.setJets(jets);
157
158 MuonCollection muons;
159 muons.push_back(badMuon);
160 ttbarEvent.setMuons(muons);
161 ttbarEvent.setMET(met);
162 ttbarEvent.setDataType(DataType::ttbar);
163 ttbarEvent.setTracks(moreThan10TracksHighPurity);
164 ttbarEvent.setBeamScrapingVeto(false);
165 assert(ttbarEvent.passesScrapingFilter());
166 assert(ttbarEvent.passesHighLevelTrigger());
167 assert(ttbarEvent.hasOneGoodPrimaryVertex());
168 assert(ttbarEvent.hasOnlyOneGoodIsolatedElectron());
169 assert(ttbarEvent.hasAtLeastFourGoodJets());
170 assert(ttbarEvent.isolatedElectronDoesNotComeFromConversion());
171 assert(ttbarEvent.isolatedElectronNotTaggedAsFromConversion());
172 }
173
174 void setUpGoodZEvent() {
175 VertexCollection vertices;
176 vertices.push_back(goodVertex);
177 goodZEvent.setVertices(vertices);
178 ElectronCollection electrons;
179 electrons.push_back(goodIsolatedElectron);
180 electrons.push_back(goodIsolatedElectron2);
181 electrons.push_back(badElectron);
182 goodZEvent.setElectrons(electrons);
183
184 JetCollection jets;
185 jets.push_back(goodJet);
186 jets.push_back(goodJet);
187 jets.push_back(goodJet);
188 jets.push_back(goodBJet);
189 goodZEvent.setJets(jets);
190 // goodZEvent.setHLT_Emulated_Photon15(true);
191
192 goodZEvent.setDataType(DataType::Zjets);
193 goodZEvent.setBeamScrapingVeto(false);
194 }
195
196 void setUpPoorZEvent() {
197 VertexCollection vertices;
198 vertices.push_back(goodVertex);
199 poorZEvent.setVertices(vertices);
200 ElectronCollection electrons;
201 electrons.push_back(goodIsolatedElectron);
202 electrons.push_back(goodLooseElectron);
203 electrons.push_back(badElectron);
204 poorZEvent.setElectrons(electrons);
205
206 JetCollection jets;
207 jets.push_back(goodJet);
208 jets.push_back(goodJet);
209 jets.push_back(goodJet);
210 jets.push_back(goodBJet);
211 poorZEvent.setJets(jets);
212 // poorZEvent.setHLT_Emulated_Photon15(true);
213
214 poorZEvent.setDataType(DataType::Zjets);
215 poorZEvent.setBeamScrapingVeto(false);
216 }
217
218 void setUpDiJetEvent() {
219 VertexCollection vertices;
220 vertices.push_back(badVertex);
221 DiJetEvent.setVertices(vertices);
222 JetCollection jets;
223 jets.push_back(goodJet);
224 jets.push_back(goodJet);
225 DiJetEvent.setJets(jets);
226 // DiJetEvent.setHLT_Emulated_Photon15(false);
227
228 DiJetEvent.setDataType(DataType::QCD_BCtoE_Pt80to170);
229 DiJetEvent.setBeamScrapingVeto(false);
230 }
231
232 void setUpDiJetEventWithConversion() {
233 VertexCollection vertices;
234 vertices.push_back(goodVertex);
235 DiJetEventWithConversion.setVertices(vertices);
236 JetCollection jets;
237 jets.push_back(goodJet);
238 jets.push_back(goodJet);
239 DiJetEventWithConversion.setJets(jets);
240 // DiJetEventWithConversion.setHLT_Emulated_Photon15(false);
241
242 ElectronCollection electrons;
243 electrons.push_back(electronFromConversion);
244 DiJetEventWithConversion.setElectrons(electrons);
245 DiJetEventWithConversion.setDataType(DataType::QCD_EMEnriched_Pt20to30);
246 DiJetEventWithConversion.setBeamScrapingVeto(false);
247 }
248
249 void setUpMuonEvent() {
250 VertexCollection vertices;
251 vertices.push_back(goodVertex);
252 muonEvent.setVertices(vertices);
253 ElectronCollection electrons;
254 electrons.push_back(goodIsolatedElectron);
255 electrons.push_back(badElectron);
256 muonEvent.setElectrons(electrons);
257 JetCollection jets;
258 jets.push_back(goodJet);
259 jets.push_back(goodJet);
260 jets.push_back(goodBJet);
261 jets.push_back(goodBJet);
262 jets.push_back(badJet);
263 muonEvent.setJets(jets);
264 // muonEvent.setHLT_Emulated_Photon15(true);
265
266 MuonCollection muons;
267 muons.push_back(goodIsolatedMuon);
268 muonEvent.setMuons(muons);
269 muonEvent.setDataType(DataType::Zjets);
270 }
271
272 void setUpTracks() {
273 TrackPointer trackHighPurity = TrackPointer(new Track(1., 0., 20., 6.));
274 TrackPointer trackLowPurity = TrackPointer(new Track(1., 0., 20., 6.));
275 trackHighPurity->setHighPurity(true);
276 trackLowPurity->setHighPurity(false);
277 for (unsigned int index = 0; index < 20; ++index) {
278 if (index < 8)
279 lessThan10Tracks.push_back(trackHighPurity);
280 if (index < 2) {
281 moreThan10TracksMixedPurity_H.push_back(trackLowPurity);
282 moreThan10TracksMixedPurity_L.push_back(trackHighPurity);
283 } else {
284 moreThan10TracksMixedPurity_H.push_back(trackHighPurity);
285 moreThan10TracksMixedPurity_L.push_back(trackLowPurity);
286 }
287
288 moreThan10TracksHighPurity.push_back(trackHighPurity);
289 moreThan10TracksLowPurity.push_back(trackLowPurity);
290 }
291 }
292
293 void setUpCustomEvent() {
294 customEvent.useCustomReturnValues = true;
295 customEvent.passConversion = false;
296 customEvent.passElectronCut = true;
297 customEvent.passHLT = true;
298 customEvent.passPV = true;
299 customEvent.passScraping = true;
300 customEvent.passes1JetCut = true;
301 customEvent.passes2JetCut = true;
302 customEvent.passes3JetCut = true;
303 customEvent.passes4JetCut = true;
304 customEvent.passesMuon = true;
305 customEvent.passesZveto = true;
306 customEvent.passConversionPartnerTrack = true;
307 assert(customEvent.passesScrapingFilter());
308 assert(customEvent.passesSelectionStep(TTbarEPlusJetsSelection::FilterOutScraping));
309 assert(customEvent.passesHighLevelTrigger());
310 assert(customEvent.passesSelectionStep(TTbarEPlusJetsSelection::HighLevelTrigger));
311 assert(customEvent.isolatedElectronDoesNotComeFromConversion() == false);
312 assert(customEvent.isolatedElectronNotTaggedAsFromConversion());
313 assert(customEvent.hasOneGoodPrimaryVertex());
314 assert(customEvent.hasOnlyOneGoodIsolatedElectron());
315 assert(customEvent.hasNoIsolatedMuon());
316 assert(customEvent.hasAtLeastOneGoodJet());
317 assert(customEvent.hasAtLeastTwoGoodJets());
318 assert(customEvent.hasAtLeastThreeGoodJets());
319 assert(customEvent.hasAtLeastFourGoodJets());
320 assert(customEvent.isNotAZBosonEvent());
321
322 assert(customEvent.useCustomReturnValues);
323 }
324
325 public:
326 void testEventConstructor() {
327 TopPairEventCandidate candidate = TopPairEventCandidate(ttbarEvent);
328 ASSERT_EQUAL(2, candidate.Electrons().size());
329 ASSERT_EQUAL(1, candidate.GoodIsolatedElectrons().size());
330 ASSERT_EQUAL(5, candidate.Jets().size());
331 ASSERT_EQUAL(4, candidate.GoodJets().size());
332 }
333
334 void testPassesHLT() {
335 ASSERT(ttbarEvent.passesHighLevelTrigger());
336 }
337
338 // void testDoesNotPassHLT() {
339 // DiJetEvent.setDataType(DataType::DATA);
340 // ASSERT_EQUAL(false, DiJetEvent.passesHighLevelTrigger());
341 // }
342
343 void testDoesHaveGoodPV() {
344 ASSERT(ttbarEvent.hasOneGoodPrimaryVertex());
345 }
346
347 void testDoesntHaveGoodPV() {
348 ASSERT(DiJetEvent.hasOneGoodPrimaryVertex() == false);
349 }
350
351 void testHasOnlyOneGoodIsolatedElectron() {
352 ASSERT(ttbarEvent.hasOnlyOneGoodIsolatedElectron());
353 }
354
355 void testHasNotOnlyOneGoodIsolatedElectron() {
356 ASSERT(goodZEvent.hasOnlyOneGoodIsolatedElectron() == false);
357 ASSERT(DiJetEvent.hasOnlyOneGoodIsolatedElectron() == false);
358 }
359
360 void testIsolatedElectronNotFromConversion() {
361 ASSERT(ttbarEvent.isolatedElectronDoesNotComeFromConversion());
362 }
363
364 void testIsolatedElectronFromConversion() {
365 ASSERT(DiJetEventWithConversion.isolatedElectronDoesNotComeFromConversion() == false);
366 }
367
368 void testNoElectronInEventReturnsFalse() {
369 ASSERT(DiJetEvent.isolatedElectronDoesNotComeFromConversion() == false);
370 }
371
372 void testHasAtLeastOneGoodJet() {
373 ASSERT(ttbarEvent.hasAtLeastOneGoodJet());
374 }
375
376 void testHasNoGoodJets() {
377 ASSERT(emptyEvent.hasAtLeastOneGoodJet() == false);
378 }
379
380 void testHasAtLeastTwoGoodjet() {
381 ASSERT(ttbarEvent.hasAtLeastTwoGoodJets());
382 }
383
384 void testHasNoTwoGoodJets() {
385 ASSERT(emptyEvent.hasAtLeastTwoGoodJets() == false);
386 }
387
388 void testHasAtLeastThreeGoodJets() {
389 ASSERT(ttbarEvent.hasAtLeastThreeGoodJets());
390 }
391
392 void testHasNoThreeGoodJets() {
393 ASSERT(DiJetEvent.hasAtLeastThreeGoodJets() == false);
394 }
395
396 void testHasAtFourThreeGoodJets() {
397 ASSERT(ttbarEvent.hasAtLeastThreeGoodJets());
398 }
399
400 void testHasNoFourThreeGoodJets() {
401 ASSERT(DiJetEvent.hasAtLeastThreeGoodJets() == false);
402 }
403
404 void testTTbarIsNotAZBosonEvent() {
405 ASSERT(ttbarEvent.isNotAZBosonEvent());
406 }
407
408 void testGoodZIsAZBosonEvent() {
409 ASSERT_EQUAL(false, goodZEvent.isNotAZBosonEvent());
410 }
411
412 void testPoorZIsAZBosonEvent() {
413 ASSERT_EQUAL(false, poorZEvent.isNotAZBosonEvent());
414 }
415
416 void testDiJetIsNotAZBosonEvent() {
417 ASSERT_EQUAL(true, DiJetEvent.isNotAZBosonEvent());
418 }
419
420 void testTTbarEventPassesMuonVeto() {
421 ASSERT_EQUAL(true, ttbarEvent.hasNoIsolatedMuon());
422 }
423
424 void testMuonEventDoesnPassMuonVeto() {
425 ASSERT_EQUAL(false, muonEvent.hasNoIsolatedMuon());
426 }
427
428 void testEventPasses0Step() {
429 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::FilterOutScraping));
430 }
431
432 void testEventPasses0StepWithLowNumberOfTracks() {
433 ttbarEvent.setTracks(lessThan10Tracks);
434 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::FilterOutScraping));
435 }
436
437 void testEventPasses0StepWithMixedTracks_H() {
438 ttbarEvent.setTracks(moreThan10TracksMixedPurity_H);
439 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::FilterOutScraping));
440 }
441
442 void testEventPasses0StepWithMixedTracks_L() {
443 ttbarEvent.setTracks(moreThan10TracksMixedPurity_L);
444 ASSERT_EQUAL(false, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::FilterOutScraping));
445 }
446
447 void testEventFails0StepWithLowPurityTracks() {
448 ttbarEvent.setTracks(moreThan10TracksLowPurity);
449 ASSERT_EQUAL(false, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::FilterOutScraping));
450 }
451
452 void testEventPasses1stStep() {
453 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::HighLevelTrigger));
454 }
455
456 void testEventPasses2ndStep() {
457 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::GoodPrimaryvertex));
458 }
459
460 void testEventPasses3ndStep() {
461 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::OneIsolatedElectron));
462 }
463
464 void testEventPasses4thStep() {
465 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::ConversionRejection));
466 }
467
468 void testEventPasses5thStep() {
469 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::LooseMuonVeto));
470 }
471
472 void testEventPasses6thStep() {
473 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::AtLeastFourGoodJets));
474 }
475
476 void testEventPasses7thStep() {
477 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStep(TTbarEPlusJetsSelection::Zveto));
478 }
479
480 void testTTbarEventPassesNStep() {
481 ASSERT_EQUAL(true, ttbarEvent.passesSelectionStepUpTo(TTbarEPlusJetsSelection::Zveto));
482 }
483
484 void testPoorZEventPassesUpToStep() {
485 assert(poorZEvent.passesScrapingFilter());
486 assert(poorZEvent.passesHighLevelTrigger());
487 assert(poorZEvent.hasOneGoodPrimaryVertex());
488 cout << poorZEvent.GoodElectrons().size() << endl;
489 cout << poorZEvent.GoodIsolatedElectrons().size() << endl;
490 assert(poorZEvent.hasOnlyOneGoodIsolatedElectron());
491 assert(poorZEvent.isolatedElectronDoesNotComeFromConversion());
492 ASSERT_EQUAL(true, poorZEvent.passesSelectionStepUpTo(TTbarEPlusJetsSelection::LooseMuonVeto));
493 }
494
495 void testPoorZEventDoesntPassUpToStep() {
496 ASSERT_EQUAL(false, poorZEvent.passesSelectionStepUpTo(TTbarEPlusJetsSelection::Zveto));
497 }
498
499 void testPassesFullTTbarSelection() {
500 ASSERT_EQUAL(true, ttbarEvent.passesFullTTbarEPlusJetSelection());
501 }
502
503 void testFailsFullTTbarSelection() {
504 ASSERT_EQUAL(false, DiJetEvent.passesFullTTbarEPlusJetSelection());
505 }
506
507 void testComputeNeutrinoPzsWithoutMETThrowsExpeption() {
508 TopPairEventCandidate cand = TopPairEventCandidate();
509 VertexCollection vertices;
510 vertices.push_back(goodVertex);
511 cand.setVertices(vertices);
512 ElectronPointer electron(new Electron(40, -40, 0, 0));
513 electron->setHcalIsolation(0);
514 electron->setEcalIsolation(0);
515 electron->setTrackerIsolation(0);
516 electron->setNumberOfMissingInnerLayerHits(0);
517 ElectronCollection eCollection;
518 eCollection.push_back(electron);
519 cand.setElectrons(eCollection);
520 ASSERT_THROWS(cand.computeNeutrinoPz(),ReconstructionException);
521 }
522
523 void testComputeNeutrinoPzsIsolatedElectronThrowsExpeption() {
524 TopPairEventCandidate cand = TopPairEventCandidate();
525 VertexCollection vertices;
526 vertices.push_back(goodVertex);
527 cand.setVertices(vertices);
528 ElectronPointer electron(new Electron(40, -40, 0, 0));
529 electron->setHcalIsolation(50);
530 electron->setEcalIsolation(50);
531 electron->setTrackerIsolation(50);
532 electron->setNumberOfMissingInnerLayerHits(0);
533 ElectronCollection eCollection;
534 eCollection.push_back(electron);
535 cand.setElectrons(eCollection);
536 ASSERT_THROWS(cand.computeNeutrinoPz(),ReconstructionException);
537 }
538
539 void testComputeNeutrinoPzs() {
540 METPointer met(new MET(40, 0));
541 ElectronPointer electron(new Electron(40, -40, 0, 0));
542 electron->setHcalIsolation(0);
543 electron->setEcalIsolation(0);
544 electron->setTrackerIsolation(0);
545 electron->setNumberOfMissingInnerLayerHits(0);
546 electron->setD0_wrtBeamSpot(0);
547 electron->setSuperClusterEta(0);
548 electron->setDEtaIn(0);
549 electron->setDPhiIn(0);
550 electron->setSigmaIEtaIEta(0);
551 electron->setHadOverEm(0);
552
553 DummyTTbarEvent cand = DummyTTbarEvent();
554 cand.setMET(met);
555 cand.setElectronFromW(electron);
556 ASSERT_EQUAL(sqrt(80*80-80.389*80.389), cand.computeNeutrinoPz().at(0));
557 ASSERT_EQUAL(sqrt(80*80-80.389*80.389), cand.computeNeutrinoPz().at(1));
558 }
559
560 void testReconstructTopEventUsingChiWithNotEnoughJetsThrowsException() {
561 METPointer met(new MET(40, 0));
562 ElectronCollection eCollection;
563 eCollection.push_back(goodIsolatedElectron);
564 TopPairEventCandidate cand = TopPairEventCandidate();
565 VertexCollection vertices;
566 vertices.push_back(goodVertex);
567 cand.setVertices(vertices);
568 cand.setMET(met);
569 cand.setElectrons(eCollection);
570 ASSERT_THROWS(cand.reconstructTTbar(goodIsolatedElectron),ReconstructionException);
571 }
572
573 void testNMinus1CutsPositive() {
574 ASSERT_EQUAL(true, customEvent.passesNMinus1(TTbarEPlusJetsSelection::ConversionRejection));
575 }
576
577 void testNMinus1CutsNegative() {
578 ASSERT_EQUAL(false, customEvent.passesNMinus1(TTbarEPlusJetsSelection::HighLevelTrigger));
579 }
580 };
581
582 extern cute::suite make_suite_TestTopPairEventCandidate() {
583 cute::suite s;
584 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testNMinus1CutsPositive));
585 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testNMinus1CutsNegative));
586 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventConstructor));
587
588 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testPassesHLT));
589 // s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testDoesNotPassHLT));
590 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testDoesHaveGoodPV));
591 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testDoesntHaveGoodPV));
592 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasOnlyOneGoodIsolatedElectron));
593 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasNotOnlyOneGoodIsolatedElectron));
594 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testIsolatedElectronNotFromConversion));
595 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testIsolatedElectronFromConversion));
596 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testNoElectronInEventReturnsFalse));
597
598 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasAtLeastOneGoodJet));
599 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasNoGoodJets));
600 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasAtLeastTwoGoodjet));
601 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasNoTwoGoodJets));
602 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasAtLeastThreeGoodJets));
603 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasNoThreeGoodJets));
604 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasAtFourThreeGoodJets));
605 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testHasNoFourThreeGoodJets));
606
607 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testTTbarIsNotAZBosonEvent));
608 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testGoodZIsAZBosonEvent));
609 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testPoorZIsAZBosonEvent));
610 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testDiJetIsNotAZBosonEvent));
611
612 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testTTbarEventPassesMuonVeto));
613 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testMuonEventDoesnPassMuonVeto));
614
615 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses0Step));
616 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses0StepWithLowNumberOfTracks));
617 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventFails0StepWithLowPurityTracks));
618 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses0StepWithMixedTracks_H));
619 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses0StepWithMixedTracks_L));
620 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses1stStep));
621 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses2ndStep));
622 // s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses2ndStepInRealData));
623 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses3ndStep));
624 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses4thStep));
625 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses5thStep));
626 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses6thStep));
627 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testEventPasses7thStep));
628
629 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testTTbarEventPassesNStep));
630 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testPoorZEventPassesUpToStep));
631 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testPoorZEventDoesntPassUpToStep));
632 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testPassesFullTTbarSelection));
633 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testFailsFullTTbarSelection));
634 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testComputeNeutrinoPzsWithoutMETThrowsExpeption));
635 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testComputeNeutrinoPzsIsolatedElectronThrowsExpeption));
636 s.push_back(CUTE_SMEMFUN(TestTopPairEventCandidate, testComputeNeutrinoPzs));
637 s.push_back(
638 CUTE_SMEMFUN(TestTopPairEventCandidate, testReconstructTopEventUsingChiWithNotEnoughJetsThrowsException));
639
640 return s;
641 }