def setUp(self):
self.v2_spe_file = SpeImage()
self.v2_spe_file.read(self.v2_spe_filename)
self.v3_spe_file = SpeImage()
self.v3_spe_file.read(self.v3_spe_filename)
self.v2_converted_spe_file = SpeImage()
self.v2_converted_spe_file.read(self.v2_converted_spe_filename)
def test_roi(self):
self.assertEqual(self.v3_spe_file.header['roi'], (0, 1024, 0, 100))
self.v3_custom_region = SpeImage()
self.v3_custom_region.read(self.v3_custom_roi_filename)
self.assertEqual(self.v3_custom_region.header['roi'], (100, 600, 10, 60))
self.assertEqual(len(self.v3_custom_region.header['x_calibration']),
self.v3_custom_region.header['x_dim'])
def test_multiple_frames(self):
self.v3_2frames_file = SpeImage()
self.v3_2frames_file.read(self.v3_2frames_filename)
self.assertEqual(self.v3_2frames_file.data.shape, (255, 1024))
frame1 = self.v3_2frames_file.data
self.v3_2frames_file.read(self.v3_2frames_filename, 1)
frame2 = self.v3_2frames_file.data
self.assertFalse(numpy.array_equal(frame1, frame2))
self.assertEqual(frame1.shape, frame2.shape)
