def setUp(self): self.img_model = ImgModel() self.mask_model = MaskModel() self.calibration_model = CalibrationModel(self.img_model) self.calibration_model._calibrants_working_dir = os.path.join( data_path, 'calibrants') self.calibration_model.integrate_1d = MagicMock() self.calibration_model.integrate_2d = MagicMock() self.calibration_widget = CalibrationWidget() self.working_dir = {} self.calibration_controller = CalibrationController( working_dir=self.working_dir, img_model=self.img_model, mask_model=self.mask_model, widget=self.calibration_widget, calibration_model=self.calibration_model)