|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#include "cppunit/extensions/HelperMacros.h"
#include <algorithm>
#include <iterator>
#include <iostream>
#include "DataFormats/Common/interface/Association.h"
#include "DataFormats/Common/interface/RefProd.h"
#include "DataFormats/Common/interface/TestHandle.h"
using namespace edm;
class testAssociationNew : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(testAssociationNew);
CPPUNIT_TEST(checkAll);
CPPUNIT_TEST_SUITE_END();
typedef std::vector<double> CVal;
typedef std::vector<int> CKey1;
typedef std::vector<float> CKey2;
public:
testAssociationNew();
void setUp() {}
void tearDown() {}
void checkAll();
void test(const edm::Association<CVal> &);
CVal k;
CKey1 v1;
CKey2 v2;
edm::TestHandle<CVal> handleV;
edm::TestHandle<CKey1> handleK1;
edm::TestHandle<CKey2> handleK2;
std::vector<int> w1, w2;
};
CPPUNIT_TEST_SUITE_REGISTRATION(testAssociationNew);
testAssociationNew::testAssociationNew() {
k.push_back(1.1);
k.push_back(2.2);
k.push_back(3.3);
ProductID const pidV(1, 1);
handleV = edm::TestHandle<CVal>(&k, pidV);
v1.push_back(1);
v1.push_back(2);
v1.push_back(3);
v1.push_back(4);
ProductID const pidK1(1, 2);
handleK1 = edm::TestHandle<CKey1>(&v1, pidK1);
v2.push_back(10.);
v2.push_back(20.);
v2.push_back(30.);
v2.push_back(40.);
v2.push_back(50.);
ProductID const pidK2(1, 3);
handleK2 = edm::TestHandle<CKey2>(&v2, pidK2);
const int ww1[4] = {2, 1, 0, 2};
w1.resize(4);
std::copy(ww1, ww1 + 4, w1.begin());
const int ww2[5] = {1, 0, 2, 1, -1};
w2.resize(5);
std::copy(ww2, ww2 + 5, w2.begin());
}
void testAssociationNew::checkAll() {
{
edm::Association<CVal> assoc(handleV);
edm::Association<CVal>::Filler filler(assoc);
filler.insert(handleK1, w1.begin(), w1.end());
filler.insert(handleK2, w2.begin(), w2.end());
filler.fill();
test(assoc);
}
{
edm::Association<CVal> assoc(handleV);
edm::Association<CVal>::Filler filler1(assoc);
filler1.insert(handleK1, w1.begin(), w1.end());
filler1.fill();
edm::Association<CVal>::Filler filler2(assoc);
filler2.insert(handleK2, w2.begin(), w2.end());
filler2.fill();
test(assoc);
}
{
edm::Association<CVal> assoc1(handleV);
edm::Association<CVal>::Filler filler1(assoc1);
filler1.insert(handleK1, w1.begin(), w1.end());
filler1.fill();
edm::Association<CVal> assoc2(handleV);
edm::Association<CVal>::Filler filler2(assoc2);
filler2.insert(handleK2, w2.begin(), w2.end());
filler2.fill();
edm::Association<CVal> assoc = assoc1 + assoc2;
test(assoc);
}
}
void testAssociationNew::test(const edm::Association<CVal> &assoc) {
CPPUNIT_ASSERT(!assoc.contains(ProductID(1, 1)));
CPPUNIT_ASSERT(assoc.contains(ProductID(1, 2)));
CPPUNIT_ASSERT(assoc.contains(ProductID(1, 3)));
CPPUNIT_ASSERT(!assoc.contains(ProductID(1, 4)));
edm::Ref<CVal> r1 = assoc[edm::Ref<CKey1>(handleK1, 0)];
edm::Ref<CVal> r2 = assoc[edm::Ref<CKey1>(handleK1, 1)];
edm::Ref<CVal> r3 = assoc[edm::Ref<CKey1>(handleK1, 2)];
edm::Ref<CVal> r4 = assoc[edm::Ref<CKey1>(handleK1, 3)];
CPPUNIT_ASSERT(r1.isNonnull());
CPPUNIT_ASSERT(r2.isNonnull());
CPPUNIT_ASSERT(r3.isNonnull());
CPPUNIT_ASSERT(r4.isNonnull());
CPPUNIT_ASSERT(*r1 == k[w1[0]]);
CPPUNIT_ASSERT(*r2 == k[w1[1]]);
CPPUNIT_ASSERT(*r3 == k[w1[2]]);
CPPUNIT_ASSERT(*r4 == k[w1[3]]);
edm::Ref<CVal> s1 = assoc[edm::Ref<CKey2>(handleK2, 0)];
edm::Ref<CVal> s2 = assoc[edm::Ref<CKey2>(handleK2, 1)];
edm::Ref<CVal> s3 = assoc[edm::Ref<CKey2>(handleK2, 2)];
edm::Ref<CVal> s4 = assoc[edm::Ref<CKey2>(handleK2, 3)];
edm::Ref<CVal> s5 = assoc[edm::Ref<CKey2>(handleK2, 4)];
CPPUNIT_ASSERT(s1.isNonnull());
CPPUNIT_ASSERT(s2.isNonnull());
CPPUNIT_ASSERT(s3.isNonnull());
CPPUNIT_ASSERT(s4.isNonnull());
CPPUNIT_ASSERT(s5.isNull());
CPPUNIT_ASSERT(*s1 == k[w2[0]]);
CPPUNIT_ASSERT(*s2 == k[w2[1]]);
CPPUNIT_ASSERT(*s3 == k[w2[2]]);
CPPUNIT_ASSERT(*s4 == k[w2[3]]);
CPPUNIT_ASSERT(assoc.size() == w1.size() + w2.size());
}
|