Belle II Software  release-08-01-10
SingleMatchSelector.test.cc
1 /**************************************************************************
2  * basf2 (Belle II Analysis Software Framework) *
3  * Author: The Belle II Collaboration *
4  * *
5  * See git log for contributors and copyright holders. *
6  * This file is licensed under LGPL-3.0, see LICENSE.md. *
7  **************************************************************************/
8 
9 #include <tracking/trackFindingCDC/collectors/selectors/SingleMatchSelector.h>
10 
11 #include <tracking/trackFindingCDC/numerics/WeightComperator.h>
12 
13 #include <gtest/gtest.h>
14 
15 using namespace Belle2;
16 using namespace TrackFindingCDC;
17 
18 namespace {
20  TEST(TrackFindingCDCTest, single_match_selector)
21  {
22  SingleMatchSelector<int, double> selector;
23 
24  int a = 1, b = 2;
25  double c = 3, d = 4, e = 5;
26 
27  std::vector<WeightedRelation<int, const double>> relations = {
28  WeightedRelation<int, const double>(&a, 1.1, &c),
29  WeightedRelation<int, const double>(&a, 1.0, &d),
30  WeightedRelation<int, const double>(&b, 0.8, &c),
31  WeightedRelation<int, const double>(&b, 0.6, &e)
32  };
33 
34  std::sort(relations.begin(), relations.end());
35  selector.apply(relations);
36 
37  std::sort(relations.begin(), relations.end(), GreaterWeight());
38 
39  ASSERT_EQ(relations.size(), 2);
40  ASSERT_EQ(relations[0].getWeight(), 1.0);
41  ASSERT_EQ(relations[0].getFrom(), &a);
42  ASSERT_EQ(relations[0].getTo(), &d);
43  ASSERT_EQ(relations[1].getWeight(), 0.6);
44  ASSERT_EQ(relations[1].getFrom(), &b);
45  ASSERT_EQ(relations[1].getTo(), &e);
46 
47  }
48 
50  TEST(TrackFindingCDCTest, single_best_match_selector)
51  {
52  SingleMatchSelector<int, double> selector;
53  selector.setUseOnlySingleBestCandidate(false);
54 
55  int a = 1, b = 2, c = 3;
56  double d = 4, e = 5, f = 6;
57 
58  std::vector<WeightedRelation<int, const double>> relations = {
59  WeightedRelation<int, const double>(&a, 1.1, &d),
60  WeightedRelation<int, const double>(&a, 1.0, &e),
61  WeightedRelation<int, const double>(&b, 0.8, &d),
62  WeightedRelation<int, const double>(&b, 0.6, &f),
63  WeightedRelation<int, const double>(&c, 0.6, &d),
64  };
65 
66  std::sort(relations.begin(), relations.end());
67  selector.apply(relations);
68 
69  std::sort(relations.begin(), relations.end(), GreaterWeight());
70 
71  ASSERT_EQ(relations.size(), 3);
72  ASSERT_EQ(relations[0].getWeight(), 1.1);
73  ASSERT_EQ(relations[0].getFrom(), &a);
74  ASSERT_EQ(relations[0].getTo(), &d);
75  ASSERT_EQ(relations[1].getWeight(), 1.0);
76  ASSERT_EQ(relations[1].getFrom(), &a);
77  ASSERT_EQ(relations[1].getTo(), &e);
78  ASSERT_EQ(relations[2].getWeight(), 0.6);
79  ASSERT_EQ(relations[2].getFrom(), &b);
80  ASSERT_EQ(relations[2].getTo(), &f);
81  }
82 
84  struct CustomComparer {
85  bool operator()(const double* lhs, const double* rhs) const
86  {
87  return (static_cast<int>(*lhs) % 2) < (static_cast<int>(*rhs) % 2);
88  }
89  };
90 
92  TEST(TrackFindingCDCTest, single_match_selector_with_comparer)
93  {
94  SingleMatchSelector<int, double, CustomComparer> selector;
95 
96  int a = 1, b = 2;
97  double c = 3, d = 4, e = 5;
98 
99  std::vector<WeightedRelation<int, const double>> relations = {
100  WeightedRelation<int, const double>(&a, 1.1, &c),
101  WeightedRelation<int, const double>(&a, 1.0, &d),
102  WeightedRelation<int, const double>(&b, 0.8, &c),
103  WeightedRelation<int, const double>(&b, 0.6, &e)
104  };
105 
106  std::sort(relations.begin(), relations.end());
107  selector.apply(relations);
108 
109  std::sort(relations.begin(), relations.end(), GreaterWeight());
110 
111  ASSERT_EQ(relations.size(), 1);
112  ASSERT_EQ(relations[0].getWeight(), 1.0);
113  ASSERT_EQ(relations[0].getFrom(), &a);
114  ASSERT_EQ(relations[0].getTo(), &d);
115  }
116 }
Belle2::TrackFindingCDC::SingleMatchSelector
Selector to remove all relations in the list, which share the same collection item - except one in ca...
Definition: SingleMatchSelector.h:45
Belle2::TrackFindingCDC::WeightedRelation
Type for two related objects with a weight.
Definition: WeightedRelation.h:26
Belle2::TEST
TEST(TestgetDetectorRegion, TestgetDetectorRegion)
Test Constructors.
Definition: utilityFunctions.cc:25
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:17
Belle2::TrackFindingCDC::BinaryJoin
Functor factory turning a binary functor and two functors into a new functor which executes the binar...
Definition: Functional.h:127