1 #include <mdst/dataobjects/Track.h>
2 #include <mdst/dataobjects/TrackFitResult.h>
3 #include <framework/gearbox/Const.h>
4 #include <framework/datastore/StoreArray.h>
5 #include <gtest/gtest.h>
30 const TVector3 dummyVector3;
31 const TMatrixDSym dummyMatrix(6);
33 const float pValue = 1.;
34 const float bField = 1.5;
36 const auto newFitRes = trackFitResults.
appendNew(dummyVector3, dummyVector3, dummyMatrix, charge, particeType, pValue,
47 DataStore::Instance().setInitializeActive(
true);
49 myResults.registerInDataStore();
53 const auto myElectron =
addDummyTrack(myResults, Const::electron);
56 EXPECT_EQ(myPion->getArrayIndex(), 0);
57 EXPECT_EQ(myKaon->getArrayIndex(), 1);
58 EXPECT_EQ(myElectron->getArrayIndex(), 2);
59 EXPECT_EQ(myMuon->getArrayIndex(), 3);
83 EXPECT_EQ(allResults.size(), 3);
86 auto countPion = std::count_if(allResults.begin(), allResults.end(),
87 [myPion](std::pair<Const::ChargedStable, const TrackFitResult*> fitPair)
88 {return (fitPair.first == Const::pion) && (fitPair.second->getArrayIndex() == myPion->getArrayIndex());});
89 auto countMuon = std::count_if(allResults.begin(), allResults.end(),
90 [myMuon](std::pair<Const::ChargedStable, const TrackFitResult*> fitPair)
91 {return (fitPair.first == Const::muon) && (fitPair.second->getArrayIndex() == myMuon->getArrayIndex());});
92 auto countElectron = std::count_if(allResults.begin(), allResults.end(),
93 [](std::pair<Const::ChargedStable, const TrackFitResult*> fitPair)
94 {return fitPair.first == Const::electron;});
95 auto countKaon = std::count_if(allResults.begin(), allResults.end(),
96 [myKaon](std::pair<Const::ChargedStable, const TrackFitResult*> fitPair)
97 {return (fitPair.first == Const::kaon) && (fitPair.second->getArrayIndex() == myKaon->getArrayIndex());});
99 EXPECT_EQ(countPion, 1);
100 EXPECT_EQ(countMuon, 1);
101 EXPECT_EQ(countElectron, 0);
102 EXPECT_EQ(countKaon, 1);
106 Track trackQITest1(0.5);
116 DataStore::Instance().setInitializeActive(
true);
118 myResults.registerInDataStore();
122 const auto myElectron =
addDummyTrack(myResults, Const::electron);
132 EXPECT_EQ(Const::kaon.getPDGCode(), fitCloseToKaonMass->getParticleType().getPDGCode());
137 EXPECT_EQ(Const::electron.getPDGCode(), wantPionButHaveElectronFit->getParticleType().getPDGCode());
142 EXPECT_EQ(Const::electron.getPDGCode(), wantMuonButHaveElectronFit->getParticleType().getPDGCode());
147 EXPECT_EQ(Const::kaon.getPDGCode(), wantProtonButHaveKaonFit->getParticleType().getPDGCode());