11 #include <tracking/trackFindingCDC/collectors/selectors/CutSelector.h>
13 #include <tracking/trackFindingCDC/numerics/WeightComperator.h>
15 #include <gtest/gtest.h>
18 using namespace TrackFindingCDC;
22 TEST(TrackFindingCDCTest, cut_selector)
24 CutSelector<int, double> selector;
27 double b = 2, c = 3, d = 4,
e = 5;
29 std::vector<WeightedRelation<int, const double>> relations = {
30 WeightedRelation<int, const double>(&a, 1.1, &b),
31 WeightedRelation<int, const double>(&a, 1.0, &c),
32 WeightedRelation<int, const double>(&a, 0.8, &d),
33 WeightedRelation<int, const double>(&a, NAN, &e)
36 selector.setCutValue(1.0);
38 std::sort(relations.begin(), relations.end());
39 selector.apply(relations);
41 std::sort(relations.begin(), relations.end(), GreaterWeight());
43 ASSERT_EQ(relations.size(), 2);
44 ASSERT_EQ(relations[0].getWeight(), 1.1);
45 ASSERT_EQ(relations[1].getWeight(), 1.0);