# This file is part of MAUS: http://micewww.pp.rl.ac.uk:8080/projects/maus # # MAUS is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # MAUS is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with MAUS. If not, see . """Tests for MapCppTrackerRecon""" # pylint: disable = C0103 import os import json import unittest import Configuration import MAUS import maus_cpp.globals import maus_cpp.converter import maus_cpp.mice_module class MapCppTrackerReconTestCase(unittest.TestCase): # pylint: disable = R0904 """Tests for MapCppTrackerRecon""" cfg = json.loads(Configuration.Configuration().getConfigJSON()) # geom = os.getenv("MAUS_ROOT_DIR")+"/src/map/"+\ # "MapCppTrackerMCDigitization/MapCppTrackerMCDigitizationTest.dat" helical_geom = "TrackerTest.dat" straight_geom = "TrackerTest_NoField.dat" cfg['simulation_geometry_filename'] = helical_geom cfg['reconstruction_geometry_filename'] = helical_geom cfg['SciFiPRHelicalOn'] = 0 cfg['SciFiStraightOn'] = 0 @classmethod def setUpClass(cls): # pylint: disable = C0103 """Sets a mapper and configuration""" cls.mapper = MAUS.MapCppTrackerRecon() cls.test_config = "" if maus_cpp.globals.has_instance(): cls.test_config = maus_cpp.globals.get_configuration_cards() maus_cpp.globals.death() def testInit(self): """Check birth""" #success = self.mapper.birth(json.dumps(self.cfg)) #self.assertTrue(success) def testGoodStraightProcess(self): """Check that tracker recon process produces expected output with good straight track data""" if maus_cpp.globals.has_instance(): maus_cpp.globals.death() self.cfg['simulation_geometry_filename'] = self.straight_geom self.cfg['reconstruction_geometry_filename'] = self.straight_geom self.cfg['SciFiPRHelicalOn'] = 0 self.cfg['SciFiPRStraightOn'] = 1 # self.cfg['SciFiPatRecVerbosity'] = 1 # print "Flags passed: " # print self.cfg['SciFiPRHelicalOn'] # print self.cfg['SciFiPRStraightOn'] maus_cpp.globals.birth(json.dumps(self.cfg)) maus_cpp.globals.set_reconstruction_mice_modules( maus_cpp.mice_module.MiceModule(self.straight_geom)) self.mapper.birth(json.dumps(self.cfg)) # Read in a spill of mc data containing 5 straight tracks test1 = \ ('%s/tests/test_data/scifi_test_data/test_straight_digits.json' % os.environ.get("MAUS_ROOT_DIR")) fin = open(test1,'r') # Check the first spill (straights) fin.readline() fin.readline() fin.readline() data = fin.readline() result = self.mapper.process(data) spill_out = maus_cpp.converter.json_repr(result) self.assertTrue('recon_events' in spill_out) self.assertEqual(1, len(spill_out['recon_events'])) # Check the first event revt = spill_out['recon_events'][0] self.assertTrue('sci_fi_event' in revt) self.assertTrue('digits' in revt['sci_fi_event']) self.assertEqual(31, len(revt['sci_fi_event']['digits'])) self.assertTrue('clusters' in revt['sci_fi_event']) self.assertEqual(30, len(revt['sci_fi_event']['clusters'])) self.assertTrue('spacepoints' in revt['sci_fi_event']) self.assertEqual(10, len(revt['sci_fi_event']['spacepoints'])) self.assertTrue('straight_pr_tracks' in revt['sci_fi_event']) self.assertEqual(2, len(revt['sci_fi_event']['straight_pr_tracks'])) self.assertTrue('helical_pr_tracks' in revt['sci_fi_event']) self.assertEqual(0, len(revt['sci_fi_event']['helical_pr_tracks'])) maus_cpp.globals.death() def testGoodHelicalProcess(self): """Check that tracker recon process produces expected output with good helical track data""" if maus_cpp.globals.has_instance(): maus_cpp.globals.death() self.cfg['simulation_geometry_filename'] = self.helical_geom self.cfg['reconstruction_geometry_filename'] = self.helical_geom self.cfg['SciFiPRHelicalOn'] = 1 self.cfg['SciFiStraightOn'] = 0 # self.cfg['SciFiPatRecVerbosity'] = 1 maus_cpp.globals.birth(json.dumps(self.cfg)) maus_cpp.globals.set_reconstruction_mice_modules( maus_cpp.mice_module.MiceModule(self.helical_geom)) self.mapper.birth(json.dumps(self.cfg)) # Read in a spill of mc data containing 5 straight tracks # test1 = ('%s/src/map/MapCppTrackerRecon/test_helical_digits.json' % # os.environ.get("MAUS_ROOT_DIR")) test1 = ('%s/tests/test_data/scifi_test_data/test_helical_digits.json' % os.environ.get("MAUS_ROOT_DIR")) fin = open(test1,'r') # Check the first spill (helices) fin.readline() fin.readline() data = fin.readline() result = self.mapper.process(data) spill_out = maus_cpp.converter.json_repr(result) self.assertTrue('recon_events' in spill_out) self.assertEqual(1, len(spill_out['recon_events'])) # Check the first event revt = spill_out['recon_events'][0] self.assertTrue('digits' in revt['sci_fi_event']) self.assertEqual(34, len(revt['sci_fi_event']['digits'])) self.assertTrue('clusters' in revt['sci_fi_event']) self.assertEqual(30, len(revt['sci_fi_event']['clusters'])) self.assertTrue('spacepoints' in revt['sci_fi_event']) self.assertEqual(10, len(revt['sci_fi_event']['spacepoints'])) self.assertTrue('straight_pr_tracks' in revt['sci_fi_event']) self.assertEqual(0, len(revt['sci_fi_event']['straight_pr_tracks'])) self.assertTrue('helical_pr_tracks' in revt['sci_fi_event']) self.assertEqual(2, len(revt['sci_fi_event']['helical_pr_tracks'])) maus_cpp.globals.death() @classmethod def tearDownClass(cls): # pylint: disable = C0103 """Sets a mapper and configuration, and checks that we can death() MapCppTrackerRecon""" cls.mapper = MAUS.MapCppTrackerRecon() if maus_cpp.globals.has_instance(): maus_cpp.globals.death() if cls.test_config != "": maus_cpp.globals.birth(cls.test_config) # Check we death() the mapper cls.mapper.death() cls.mapper = None if __name__ == '__main__': unittest.main()