class DetectorTest(unittest.TestCase): def setUp(self): self.d = DetectorData( shape = (10,10), pixelsteps_per_buffer = 1, buffers_per_file = 1, dirpaths = NETCDF_DIR, filepattern = NETCDF_PATTERN, mca_bins = 2048, first_file_n = 1, ) def get_spectrum_first_test(self): pixel_step, row, col = (0, 0, 0) spectrum = self.d.spectrum(pixel_step, row, col) self.assertEqual(len(spectrum), 2048) self.assertEqual(spectrum.sum(), 5445) def get_spectrum_last_test(self): pixel_step, row, col = (538, 9, 9) spectrum = self.d.spectrum(pixel_step, row, col) self.assertEqual(len(spectrum), 2048) self.assertEqual(spectrum.sum(), 155276) def get_statistic_first_test(self): pixel_step, row, col = (0, 0, 0) stat = self.d.statistic(pixel_step, row, col, 'realtime') self.assertEqual(stat, 3125023)
class DetectorTest(unittest.TestCase): def setUp(self): self.d = DetectorData( shape=(10, 10), pixelsteps_per_buffer=1, buffers_per_file=1, dirpaths=NETCDF_DIR, filepattern=NETCDF_PATTERN, mca_bins=2048, first_file_n=1, ) def get_spectrum_first_test(self): pixel_step, row, col = (0, 0, 0) spectrum = self.d.spectrum(pixel_step, row, col) self.assertEqual(len(spectrum), 2048) self.assertEqual(spectrum.sum(), 5445) def get_spectrum_last_test(self): pixel_step, row, col = (538, 9, 9) spectrum = self.d.spectrum(pixel_step, row, col) self.assertEqual(len(spectrum), 2048) self.assertEqual(spectrum.sum(), 155276) def get_statistic_first_test(self): pixel_step, row, col = (0, 0, 0) stat = self.d.statistic(pixel_step, row, col, 'realtime') self.assertEqual(stat, 3125023)