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)
示例#2
0
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)