8#include <gtest/gtest.h>
10#include <analysis/VariableManager/Manager.h>
12#include <analysis/dataobjects/Particle.h>
13#include <analysis/dataobjects/RestOfEvent.h>
15#include <mdst/dataobjects/Track.h>
16#include <mdst/dataobjects/ECLCluster.h>
17#include <mdst/dataobjects/KLMCluster.h>
18#include <mdst/dataobjects/PIDLikelihood.h>
20#include "utilities/TestParticleFactory.h"
23using namespace Belle2::Variable;
28 class InclusiveVariablesTest :
public ::testing::Test {
35 StoreArray<ECLCluster> myECLClusters;
36 StoreArray<KLMCluster> myKLMClusters;
37 StoreArray<TrackFitResult> myTFRs;
38 StoreArray<Track> myTracks;
39 StoreArray<Particle> myParticles;
40 StoreArray<RestOfEvent> myROEs;
41 StoreArray<PIDLikelihood> myPIDLikelihoods;
53 TestUtilities::TestParticleFactory factory;
54 ROOT::Math::XYZVector ipposition(0, 0, 0);
55 ROOT::Math::PxPyPzEVector b0momentum(3, 0, 0, 5);
56 factory.
produceParticle(
string(
"^B0 -> [^K*0 -> [^K_S0 -> ^pi+ ^pi-] [^pi0 -> ^gamma ^gamma]] ^e+ ^e-"),
57 b0momentum, ipposition);
61 void TearDown()
override
66 TEST_F(InclusiveVariablesTest, nCompositeDaughters)
70 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 1);
72 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 1);
74 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 0);
76 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 0);
78 TEST_F(InclusiveVariablesTest, nCompositeAllGenerationDaughters)
82 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 3);
84 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 1);
86 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 1);
88 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 1);
90 TEST_F(InclusiveVariablesTest, nPhotonDaughters)
94 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 2);
96 TEST_F(InclusiveVariablesTest, nDaughterNeutralHadrons)
100 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 0);
102 TEST_F(InclusiveVariablesTest, nChargedDaughters)
106 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 4);
108 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 2);
110 EXPECT_EQ(std::get<int>(var->function(myParticles[9])), 2);
113 TEST_F(InclusiveVariablesTest, daughterAverageOf)
117 EXPECT_FLOAT_EQ(std::get<double>(var->function(myParticles[9])),
float(313 + 11 - 11) / 3);
static DataStore & Instance()
Instance of singleton Store.
void setInitializeActive(bool active)
Setter for m_initializeActive.
void reset(EDurability durability)
Frees memory occupied by data store items and removes all objects from the map.
bool registerInDataStore(DataStore::EStoreFlags storeFlags=DataStore::c_WriteOut)
Register the object/array in the DataStore.
Accessor to arrays stored in the data store.
bool registerRelationTo(const StoreArray< TO > &toArray, DataStore::EDurability durability=DataStore::c_Event, DataStore::EStoreFlags storeFlags=DataStore::c_WriteOut, const std::string &namedRelation="") const
Register a relation to the given StoreArray.
const Var * getVariable(std::string name)
Get the variable belonging to the given key.
static Manager & Instance()
get singleton instance.
const Belle2::Particle * produceParticle(const std::string &decayString, const ROOT::Math::PxPyPzEVector &momentum, const ROOT::Math::XYZVector &vertex)
Main method to produce particles.
Abstract base class for different kinds of events.