← Back to branch summary

~ajdobbs/maus/rootfitter

« back to all changes in this revision

Viewing changes to src/map/MapCppGlobalPID/test_MapCppGlobalPID.py

  • Committer: Adam Dobbs
  • Date: 2017-02-28 17:58:27 UTC
  • mfrom: (1149.2.21 maus-primarychain)
  • Revision ID: phuccj@gmail.com-20170228175827-vhunrks5hzqsayb5
Merge: global recon updates

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/map/MapCppGlobalPID/test_MapCppGlobalPID.py
106
106
            self.assertEqual(4, len(gevt)) 
107
107
            self.assertTrue('tracks' in gevt)
108
108
            tracksarray = gevt['tracks']
109
 
            for i in tracksarray:
110
 
                if i['mapper_name'] == 'MapCppGlobalPID-Final':
111
 
                    track = i
 
109
            num_tracks = 0;
 
110
            for track in tracksarray:
 
111
                if track['mapper_name'] == 'MapCppGlobalPID':
112
112
                    self.assertTrue('pid' in track)
113
113
                    self.assertEqual(-13, track['pid'])
 
114
                    num_tracks += 1
 
115
            self.assertEqual(num_tracks, 3)
114
116
 
115
117
    def test_positron_PID(self):
116
118
        """Check that process can identify positrons"""
130
132
            self.assertEqual(4, len(gevt)) 
131
133
            self.assertTrue('tracks' in gevt)
132
134
            tracksarray = gevt['tracks']
133
 
            for i in tracksarray:
134
 
                if i['mapper_name'] == 'MapCppGlobalPID-Final':
135
 
                    track = i
 
135
            num_tracks = 0;
 
136
            for track in tracksarray:
 
137
                if track['mapper_name'] == 'MapCppGlobalPID':
136
138
                    self.assertTrue('pid' in track)
137
139
                    self.assertEqual(-11, track['pid'])
 
140
                    num_tracks += 1
 
141
            self.assertEqual(num_tracks, 3)
138
142
 
139
143
    def test_pion_PID(self):
140
144
        """Check that process can identify pions"""
154
158
            self.assertEqual(4, len(gevt)) 
155
159
            self.assertTrue('tracks' in gevt)
156
160
            tracksarray = gevt['tracks']
157
 
            for i in tracksarray:
158
 
                if i['mapper_name'] == 'MapCppGlobalPID-Final':
159
 
                    track = i
 
161
            num_tracks = 0;
 
162
            for track in tracksarray:
 
163
                if track['mapper_name'] == 'MapCppGlobalPID':
160
164
                    self.assertTrue('pid' in track)
161
165
                    self.assertEqual(211, track['pid'])
162
 
 
163
 
    def test_undef_PID(self):
164
 
        """Check that PID set to 0 for indistinguishable particles"""
165
 
        test7 = ('%s/src/map/MapCppGlobalPID/undef_pid_test.json' %
166
 
                 os.environ.get("MAUS_ROOT_DIR"))
167
 
        self.mapper.birth(json.dumps(self.config0))
168
 
        fin = open(test7,'r')
169
 
        for line in fin:
170
 
            result = self.mapper.process(line)
171
 
            spill_out = maus_cpp.converter.json_repr(result)
172
 
            self.assertTrue('recon_events' in spill_out)
173
 
            revtarray = spill_out['recon_events']
174
 
            self.assertEqual(1, len(revtarray))
175
 
            revt = revtarray[0]
176
 
            self.assertTrue('global_event' in revt)
177
 
            gevt = revt['global_event']
178
 
            self.assertTrue('tracks' in gevt)
179
 
            tracksarray = gevt['tracks']
180
 
            for i in tracksarray:
181
 
                if i['mapper_name'] == 'MapCppGlobalPID-Final':
182
 
                    track = i
183
 
                    self.assertTrue('pid' in track)
184
 
                    self.assertEqual(0, track['pid'])
 
166
                    num_tracks += 1
 
167
            self.assertEqual(num_tracks, 3)
185
168
 
186
169
    def test_invalid_logL(self):
187
170
        """Check that a track that returns an invalid logL does not get
202
185
            self.assertEqual(4, len(gevt)) 
203
186
            self.assertTrue('tracks' in gevt)
204
187
            tracksarray = gevt['tracks']
205
 
            self.assertEqual(3, len(gevt['tracks']))
206
 
            track = tracksarray[0]
207
 
            self.assertTrue('pid' in track)
208
 
            self.assertEqual(0, track['pid'])
 
188
            self.assertEqual(2, len(gevt['tracks']))
 
189
            num_tracks = 0;
 
190
            for track in tracksarray:
 
191
                if track['mapper_name'] == 'MapCppGlobalPID':
 
192
                    num_tracks += 1
 
193
            self.assertEqual(num_tracks, 0)
209
194
    
210
195
    @classmethod
211
196
    def tearDownClass(cls): # pylint: disable = C0103