def setUp(self):
        self.img_model = ImgModel()
        self.mask_model = MaskModel()
        self.spectrum_model = PatternModel()

        # setting up the calibration model but mocking the integration for speed
        self.calibration_model = CalibrationModel(self.img_model)
        self.calibration_model.num_points = 1000
        dummy_x = np.linspace(0, 25, 1000)
        dummy_y = np.sin(dummy_x)
        self.calibration_model.integrate_1d = mock.Mock(return_value=(dummy_x,
                                                                      dummy_y))

        self.phase_model = PhaseModel()
        self.widget = IntegrationWidget()
        self.integration_controller = IntegrationController(
            {'spectrum': data_path},
            widget=self.widget,
            img_model=self.img_model,
            mask_model=self.mask_model,
            calibration_model=self.calibration_model,
            spectrum_model=self.spectrum_model,
            phase_model=self.phase_model)
        self.image_controller = self.integration_controller.image_controller
        self.calibration_model.load(
            os.path.join(data_path, 'CeO2_Pilatus1M.poni'))
        self.img_model.load(os.path.join(data_path, 'CeO2_Pilatus1M.tif'))
Example #2
0
    def setUp(self):
        self.img_model = ImgModel()
        self.mask_model = MaskModel()
        self.spectrum_model = PatternModel()
        self.calibration_model = CalibrationModel(self.img_model)
        self.calibration_model.integrate_1d = MagicMock(
            return_value=(self.calibration_model.tth,
                          self.calibration_model.int))

        self.phase_model = PhaseModel()

        self.integration_widget = IntegrationWidget()

        self.integration_controller = IntegrationController(
            {'spectrum': data_path},
            widget=self.integration_widget,
            img_model=self.img_model,
            mask_model=self.mask_model,
            calibration_model=self.calibration_model,
            spectrum_model=self.spectrum_model,
            phase_model=self.phase_model)
        self.calibration_model.load(
            os.path.join(data_path, 'CeO2_Pilatus1M.poni'))
        self.img_model.load(os.path.join(data_path, 'CeO2_Pilatus1M.tif'))

        self.integration_spectrum_controller = self.integration_controller.spectrum_controller
        self.integration_image_controller = self.integration_controller.image_controller