~christopher-hunt08/maus/maus_scifi_recon_devel

« back to all changes in this revision

Viewing changes to tests/cpp_unit/DataStructure/SciFiStraightPRTrackTest.cc

  • Committer: Christopher Hunt
  • Date: 2015-12-14 08:48:38 UTC
  • mfrom: (697.47.179 cctag_corr)
  • Revision ID: christopher.hunt08@imperial.ac.uk-20151214084838-wjl32e152yx3ks1j
MAUS Merg Remerged

Show diffs side-by-side

added added

removed removed

Lines of Context:
41
41
  EXPECT_EQ(prtrack.get_my(), -1.0);
42
42
  EXPECT_EQ(prtrack.get_y_chisq(), -1.0);
43
43
  EXPECT_EQ(prtrack.get_tracker(), -1);
44
 
  EXPECT_EQ(prtrack.get_num_points(), -1);
 
44
  EXPECT_EQ(prtrack.get_num_points(), 0);
45
45
  EXPECT_EQ(prtrack.get_covariance().size(), 0);
46
46
}
47
47
 
48
48
TEST_F(SciFiStraightPRTrackTestDS, test_parameter_constructor) {
49
49
  std::vector<double> cov(0);
50
50
  cov.push_back(10.0);
51
 
  SciFiStraightPRTrack prtrack(0, 3, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, cov);
 
51
  SciFiStraightPRTrack prtrack(0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, cov);
52
52
  EXPECT_EQ(prtrack.get_x0(), 1.0);
53
53
  EXPECT_EQ(prtrack.get_mx(), 2.0);
54
54
  EXPECT_EQ(prtrack.get_x_chisq(), 3.0);
56
56
  EXPECT_EQ(prtrack.get_my(), 5.0);
57
57
  EXPECT_EQ(prtrack.get_y_chisq(), 6.0);
58
58
  EXPECT_EQ(prtrack.get_tracker(), 0);
59
 
  EXPECT_EQ(prtrack.get_num_points(), 3);
 
59
  EXPECT_EQ(prtrack.get_num_points(), 0);
60
60
  ASSERT_EQ(prtrack.get_covariance().size(), 1);
61
61
  EXPECT_EQ(prtrack.get_covariance()[0], 10.0);
62
62
}
71
71
  ly.set_c(4.0);
72
72
  ly.set_m(5.0);
73
73
  ly.set_chisq(6.0);
74
 
  SciFiStraightPRTrack prtrack(0, 3, lx, ly, cov);
 
74
  SciFiStraightPRTrack prtrack(0, lx, ly, cov);
75
75
  EXPECT_EQ(prtrack.get_x0(), 1.0);
76
76
  EXPECT_EQ(prtrack.get_mx(), 2.0);
77
77
  EXPECT_EQ(prtrack.get_x_chisq(), 3.0);
79
79
  EXPECT_EQ(prtrack.get_my(), 5.0);
80
80
  EXPECT_EQ(prtrack.get_y_chisq(), 6.0);
81
81
  EXPECT_EQ(prtrack.get_tracker(), 0);
82
 
  EXPECT_EQ(prtrack.get_num_points(), 3);
 
82
  EXPECT_EQ(prtrack.get_num_points(), 0);
83
83
  ASSERT_EQ(prtrack.get_covariance().size(), 1);
84
84
  EXPECT_EQ(prtrack.get_covariance()[0], 10.0);
85
85
}
87
87
TEST_F(SciFiStraightPRTrackTestDS, test_copy_constructor) {
88
88
  std::vector<double> cov(0);
89
89
  cov.push_back(10.0);
90
 
  SciFiStraightPRTrack trk1(0, 3, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, cov);
 
90
  SciFiStraightPRTrack trk1(0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, cov);
91
91
 
92
92
  SciFiSpacePoint *spoint = new SciFiSpacePoint();
93
93
  spoint->set_tracker(1);
104
104
  EXPECT_EQ(trk2.get_my(), 5.0);
105
105
  EXPECT_EQ(trk2.get_y_chisq(), 6.0);
106
106
  EXPECT_EQ(trk2.get_tracker(), 0);
107
 
  EXPECT_EQ(trk2.get_num_points(), 3);
 
107
  EXPECT_EQ(trk2.get_num_points(), 1);
108
108
  EXPECT_EQ(trk2.get_spacepoints_pointers()[0]->get_tracker(), 1);
109
109
  ASSERT_EQ(trk2.get_covariance().size(), 1);
110
110
  EXPECT_EQ(trk2.get_covariance()[0], 10.0);
113
113
TEST_F(SciFiStraightPRTrackTestDS, test_equality_operator) {
114
114
  std::vector<double> cov(0);
115
115
  cov.push_back(10.0);
116
 
  SciFiStraightPRTrack trk1(0, 3, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, cov);
 
116
  SciFiStraightPRTrack trk1(0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, cov);
117
117
 
118
118
  SciFiSpacePoint *spoint = new SciFiSpacePoint();
119
119
  spoint->set_tracker(1);
131
131
  EXPECT_EQ(trk2.get_my(), 5.0);
132
132
  EXPECT_EQ(trk2.get_y_chisq(), 6.0);
133
133
  EXPECT_EQ(trk2.get_tracker(), 0);
134
 
  EXPECT_EQ(trk2.get_num_points(), 3);
 
134
  EXPECT_EQ(trk2.get_num_points(), 1);
135
135
  EXPECT_EQ(trk2.get_spacepoints_pointers()[0]->get_tracker(), 1);
136
136
  ASSERT_EQ(trk2.get_covariance().size(), 1);
137
137
  EXPECT_EQ(trk2.get_covariance()[0], 10.0);
147
147
  int tracker = 1;
148
148
  int num_points = 5;
149
149
 
150
 
  SciFiSpacePoint *spoint = new SciFiSpacePoint();
151
 
  spoint->set_tracker(tracker);
152
150
  SciFiSpacePointPArray spoints;
153
 
  spoints.push_back(spoint);
 
151
  for (int it_p = 0; it_p < num_points; ++it_p) {
 
152
    SciFiSpacePoint *spoint = new SciFiSpacePoint();
 
153
    spoint->set_tracker(tracker);
 
154
    spoints.push_back(spoint);
 
155
  }
154
156
 
155
157
  std::vector<double> cov(0);
156
158
  cov.push_back(10.0);
164
166
  prtrack.set_my(my);
165
167
  prtrack.set_y_chisq(y_chisq);
166
168
  prtrack.set_tracker(tracker);
167
 
  prtrack.set_num_points(num_points);
168
169
  prtrack.set_spacepoints_pointers(spoints);
169
170
  prtrack.set_covariance(cov);
170
171