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'))
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