75
75
self.assertEqual(output.image_types,
76
76
self.test_cards["image_types"])
78
def test_death(self): # pylint: disable=R0201
79
"""OutputPyRootImage test death()"""
80
output = OutputPyRootImage()
84
"""OutputPyRootImage test save()"""
85
output = OutputPyRootImage()
86
output.birth(json.dumps(self.test_cards))
87
output.save(self.test_image)
88
test_path = os.path.join(self.test_cards["image_directory"],
89
"imageTestCanvas.png")
90
self.assertTrue(os.path.exists(test_path))
91
self.test_image.GetImage().SetSpillNumber(1)
92
output.save(self.test_image)
93
self.test_image.GetImage().SetSpillNumber(2)
94
output.save(self.test_image)
95
self.assertTrue(os.path.exists(test_path))
97
self.test_image.GetImage().SetSpillNumber(3)
98
output.save(self.test_image)
99
self.assertTrue(os.path.exists(test_path))
101
def test_save_end_of_run(self):
102
"""OutputPyRootImage test end of run save()"""
103
output = OutputPyRootImage()
104
output.birth(json.dumps(self.test_cards))
105
self.test_image.GetImage().SetSpillNumber(4)
106
output.save(self.test_image)
107
output.save(ROOT.MAUS.RunFooterData())
108
test_path = os.path.join(self.test_cards["end_of_run_image_directory"],
109
str(self.test_image.GetImage().GetRunNumber()),
110
"imageTestCanvas.png")
111
self.assertTrue(os.path.exists(test_path))
112
output.save(ROOT.MAUS.RunFooterData())
113
self.assertTrue(os.path.exists(test_path))
78
# def test_death(self): # pylint: disable=R0201
79
# """OutputPyRootImage test death()"""
80
# output = OutputPyRootImage()
84
# def test_save(self):
85
# """OutputPyRootImage test save()"""
86
# output = OutputPyRootImage()
87
# output.birth(json.dumps(self.test_cards))
88
# output.save(self.test_image)
89
# test_path = os.path.join(self.test_cards["image_directory"],
90
# "imageTestCanvas.png")
91
# self.assertTrue(os.path.exists(test_path))
92
# self.test_image.GetImage().SetSpillNumber(1)
93
# output.save(self.test_image)
94
# self.test_image.GetImage().SetSpillNumber(2)
95
# output.save(self.test_image)
96
# self.assertTrue(os.path.exists(test_path))
97
# os.remove(test_path)
98
# self.test_image.GetImage().SetSpillNumber(3)
99
# output.save(self.test_image)
100
# self.assertTrue(os.path.exists(test_path))
102
# def test_save_end_of_run(self):
103
# """OutputPyRootImage test end of run save()"""
104
# output = OutputPyRootImage()
105
# output.birth(json.dumps(self.test_cards))
106
# self.test_image.GetImage().SetSpillNumber(4)
107
# output.save(self.test_image)
108
# output.save(ROOT.MAUS.RunFooterData())
109
# test_path = os.path.join(\
110
# self.test_cards["end_of_run_image_directory"],
111
# str(self.test_image.GetImage().GetRunNumber()),
112
# "imageTestCanvas.png")
113
# self.assertTrue(os.path.exists(test_path))
114
# output.save(ROOT.MAUS.RunFooterData())
115
# self.assertTrue(os.path.exists(test_path))
116
118
if __name__ == "__main__":