def setUp(self): self.context = setup_context(True) self.context.data_context.instrument = 'MUSR' self.context.gui_context.update({'RebinType': 'None'}) self.view = fft_view_new.FFTView() self.model1 = fft_model.FFTModel() self.model = fft_model.FFTWrapper self.presenter = fft_presenter_new.FFTPresenter( self.view, self.model, self.context) file_path = FileFinder.findRuns('MUSR00022725.nxs')[0] ws, run, filename, _ = load_utils.load_workspace_from_filename(file_path) self.context.data_context._loaded_data.remove_data(run=run) self.context.data_context._loaded_data.add_data(run=[run], workspace=ws, filename=filename, instrument='MUSR') self.context.data_context.current_runs = [[22725]] self.context.update_current_data() test_pair = MuonPair(EXAMPLE_PAIR, 'top', 'bottom', alpha=0.75) self.context.group_pair_context.add_pair(pair=test_pair) self.context.show_all_groups() self.context.show_all_pairs() self.context.group_pair_context._selected_groups = GROUP_LIST self.context.group_pair_context._selected_pairs = [EXAMPLE_PAIR] self.view.warning_popup = mock.MagicMock()
def setUp(self): self.context = setup_context(True) self.context.data_context.instrument = 'MUSR' self.context.gui_context.update({'RebinType': 'None'}) # View self.view = mock.create_autospec(maxent_view.MaxEntView, spec_set=True) # signals # needed for connect in presenter self.view.maxEntButtonSignal = mock.Mock() self.view.cancelSignal = mock.Mock() # Default Values type(self.view).get_run = mock.PropertyMock(return_value="2435") type(self.view).num_periods = mock.PropertyMock(return_value=1) type(self.view).get_period = mock.PropertyMock(return_value="1") self.view.phase_table = "Construct" self.view.num_points = 2048 self.view.inner_iterations = 10 self.view.outer_iterations = 10 self.view.double_pulse = False self.view.lagrange_multiplier = 1 self.view.maximum_field = 1000 self.view.maximum_entropy_constant = 0.1 self.view.fit_dead_times = False self.context.data_context.num_periods = mock.Mock(return_value=1) # Presenter self.presenter = maxent_presenter.MaxEntPresenter(self.view, self.context) # make thread self.thread = mock.create_autospec(thread_model.ThreadModel)
def test_get_pair_phasequad_name(self): AnalysisDataService.clear() ConfigService['MantidOptions.InvisibleWorkspaces'] = 'True' filepath = FileFinder.findRuns('EMU00019489.nxs')[0] load_result, run_number, filename, psi_data = load_workspace_from_filename( filepath) context = setup_context() context.gui_context.update({'RebinType': 'None'}) context.data_context.instrument = 'EMU' context.data_context._loaded_data.add_data(workspace=load_result, run=[run_number], filename=filename, instrument='EMU') context.data_context.current_runs = [[run_number]] context.data_context.update_current_data() self.assertEqual( "EMU19489; PhaseQuad; test_Re; MA", get_pair_phasequad_name(context, "test_Re", "19489", False)) self.assertEqual( "EMU19489; PhaseQuad; test_Re; Rebin; MA", get_pair_phasequad_name(context, "test_Re", "19489", True))
def setUp(self): self.context = setup_context(True) self.context.data_context.instrument = 'MUSR' self.context.gui_context.update({'RebinType': 'None'}) self.model = maxent_model.MaxEntModel() self.view = maxent_view_new.MaxEntView() self.presenter = maxent_presenter_new.MaxEntPresenter( self.view, self.context) file_path = FileFinder.findRuns('MUSR00022725.nxs')[0] ws, run, filename, _ = load_utils.load_workspace_from_filename( file_path) self.context.data_context._loaded_data.remove_data(run=run) self.context.data_context._loaded_data.add_data(run=[run], workspace=ws, filename=filename, instrument='MUSR') self.context.data_context.current_runs = [[22725]] self.context.update_current_data() test_pair = MuonPair('test_pair', 'top', 'bottom', alpha=0.75) self.context.group_pair_context.add_pair(pair=test_pair) self.view.warning_popup = mock.MagicMock()
def setUp(self): self.context = setup_context() self.model = mock.Mock(spec=PlotTimeFitPaneModel) self.model.name = "data" # manually add pane self.context.plot_panes_context.add_pane(self.model.name) self.view = mock.Mock(spec=BasePaneView) self.view.warning_popup = mock.MagicMock() self.view.disable_plot_raw_option() self.view.hide_plot_type = mock.Mock() self.view.hide_plot_raw = mock.Mock() self.view.hide_tiled_by = mock.Mock() self.figure_presenter = mock.Mock( spec=PlottingCanvasPresenterInterface) self.figure_presenter.set_errors = mock.Mock() self.figure_presenter.set_plot_range = mock.Mock() self.presenter = PlotModelFitPanePresenter( view=self.view, model=self.model, context=self.context, fitting_context=self.context.fitting_context, figure_presenter=self.figure_presenter)
def setUp(self): AnalysisDataService.clear() ConfigService['MantidOptions.InvisibleWorkspaces'] = 'True' self.filepath = FileFinder.findRuns('EMU00019489.nxs')[0] self.load_result, self.run_number, self.filename, psi_data = load_workspace_from_filename( self.filepath) self.assert_(not psi_data) self.context = setup_context(True) self.context.gui_context.update({'RebinType': 'None'}) self.loaded_data = self.context.data_context._loaded_data self.data_context = self.context.data_context self.gui_context = self.context.gui_context self.group_pair_context = self.context.group_pair_context self.data_context.instrument = 'EMU' self.loaded_data.add_data(workspace=self.load_result, run=[self.run_number], filename=self.filename, instrument='EMU') self.data_context.current_runs = [[self.run_number]] self.data_context.update_current_data() self.group_pair_context.reset_group_and_pairs_to_default( self.load_result['OutputWorkspace'][0].workspace, 'EMU', '', 1)
def setUp(self): self.current_runs = [[22725]] self.context = setup_context(True) self.context.get_names_of_workspaces_to_fit = mock.MagicMock( return_value=['MUSR22725; Group; fwd; Asymmetry; #1']) self.view = WorkspaceSelectorView(self.current_runs, 'MUSR', [], True, None, self.context) self.view.list_selector_presenter = mock.MagicMock() self.context.get_names_of_workspaces_to_fit.reset_mock()
def setUp(self): context = setup_context() self.model = ModelFittingModel(context, context.model_fitting_context) self.result_table_names = ["Result1", "Result2"] self.dataset_names = ["workspace_name_A0", "workspace_name_A1", "A0_A1"] self.fit_function1 = FunctionFactory.createFunction("FlatBackground") self.fit_function2 = FunctionFactory.createFunction("LinearBackground") self.single_fit_functions = [self.fit_function1.clone(), self.fit_function1.clone(), self.fit_function2.clone()]
def setUp(self): self.context = setup_context() self.context.data_context.current_runs = [[62260]] self.context.data_context.instrument = 'MUSR' self.widget = FittingTabWidget(self.context, parent=None) self.presenter = self.widget.fitting_tab_presenter self.view = self.widget.fitting_tab_view self.presenter.model = mock.MagicMock()
def setUp(self): context = setup_context() self.model = GeneralFittingModel(context, context.fitting_context) self.dataset_names = ["Name1", "Name2"] self.fit_function = FunctionFactory.createFunction("FlatBackground") self.single_fit_functions = [self.fit_function.clone(), self.fit_function.clone()] self.simultaneous_fit_function = FunctionFactory.createInitializedMultiDomainFunction(str(self.fit_function), len(self.dataset_names))
def setUp(self): self.context = setup_context(True) self.workspace_finder = WorkspaceFinder(self.context) test_group = MuonGroup(group_name="fwd", detector_ids=[1, 2, 3, 4, 5]) self.context.group_pair_context._groups = [test_group] self.context.group_pair_context._pairs = [] self.context.group_pair_context._selected_groups = ["fwd"] self.context.group_pair_context._selected_pairs = []
def setUp(self): self.context = setup_context() self.context.data_context.current_runs = [[62260]] self.context.data_context.instrument = 'MUSR' self.widget = FittingTabWidget(self.context, parent=None) self.presenter = self.widget.fitting_tab_presenter self.view = self.widget.fitting_tab_view self.view.warning_popup = mock.MagicMock() self.presenter.model = mock.MagicMock() self.presenter.model.get_function_name.return_value = 'GausOsc'
def setUp(self): self.model = FittingTabModel(setup_context()) self.model.context.ads_observer.unsubscribe() self.model.fitting_options = { "startX": START_X, "endX": END_X, "evaluation_type": EVALUATION_TYPE, "minimiser": MINIMISER, "fit_type": "Single", "tf_asymmetry_mode": False }
def setUp(self): self.context = setup_context() self.view = mock.MagicMock() self.view.fit_table = mock.MagicMock() self.model = mock.MagicMock() self.widget = SeqFittingTabWidget(context=self.context, model=self.model, parent=None, view=self.view) self.presenter = self.widget.seq_fitting_tab_presenter self.view.is_plotting_checked.return_value = False self.view.use_initial_values_for_fits.return_value = False self.presenter.create_thread = mock.MagicMock() self.view.fit_table.get_workspace_info_from_row = mock.MagicMock(return_value=["2224;2225", "bwd;fwd;top"])
def setUp(self): self.context = setup_context() self.view = mock.MagicMock() self.view.fit_table = mock.MagicMock() self.model = mock.MagicMock() self.presenter = SeqFittingTabPresenter(self.view, self.model, self.context) self.view.is_plotting_checked.return_value = False self.view.use_initial_values_for_fits.return_value = False self.presenter.create_thread = mock.MagicMock() self.view.fit_table.get_workspace_info_from_row = mock.MagicMock( return_value=["2224;2225", "bwd;fwd;top"])
def setUp(self): self.view = PhaseTableView() self.context = setup_context() self.context.data_context.instrument = 'MUSR' self.presenter = PhaseTablePresenter(self.view, self.context) forward_group = MuonGroup(group_name="fwd", detector_ids=[1,3,5,7,9]) backward_group = MuonGroup(group_name="bwd", detector_ids=[2,4,6,8,10]) self.context.group_pair_context.add_group(forward_group) self.context.group_pair_context.add_group(backward_group) self.presenter.update_current_groups_list() self.view.warning_popup = mock.MagicMock()
def setUp(self): self._qapp = mock_widget.mockQapp() self.view = PhaseTableView() self.context = setup_context() self.context.data_context.instrument = 'MUSR' self.presenter = PhaseTablePresenter(self.view, self.context) forward_group = MuonGroup(group_name="fwd", detector_ids=[1,3,5,7,9]) backward_group = MuonGroup(group_name="bwd", detector_ids=[2,4,6,8,10]) self.context.group_pair_context.add_group(forward_group) self.context.group_pair_context.add_group(backward_group) self.presenter.update_current_groups_list() self.view.warning_popup = mock.MagicMock()
def setUp(self): self.context = setup_context(True) self.plotting_window_model = mock.MagicMock() self.view = mock.MagicMock() self.model = mock.MagicMock() self.workspace_list = [ 'MUSR62260; Group; bottom; Asymmetry; FD', 'MUSR62261; Group; bottom; Asymmetry; FD', 'FFT; Re MUSR62260; Group; fwd; Asymmetry; FD; Im MUSR62260; Group; fwd; Asymmetry; FD_Im' 'FFT; Re MUSR62260; Group; fwd; Asymmetry; FD; Im MUSR62260; Group; fwd; Asymmetry; FD_Re' 'FFT; Re MUSR62260; Group; fwd; Asymmetry; FD; Im MUSR62260; Group; fwd; Asymmetry; FD_mod' 'MUSR62260_raw_data FD; MaxEnt' ] self.presenter = HomePlotWidgetPresenter(self.view, self.model, self.context) self.presenter.get_plot_title = mock.MagicMock( return_value='MUSR62260-62261 bottom')
def _setup_presenter(self): self.context = setup_context() self.presenter = CorrectionsPresenter(self.view, self.model, self.context) self.presenter.disable_editing_notifier = mock.Mock() self.presenter.enable_editing_notifier = mock.Mock() self.presenter.asymmetry_pair_and_diff_calculations_finished_notifier = mock.Mock( ) self.presenter.dead_time_presenter.initialize_model_options = mock.Mock( ) self.presenter.dead_time_presenter.handle_ads_clear_or_remove_workspace_event = mock.Mock( ) self.presenter.dead_time_presenter.handle_instrument_changed = mock.Mock( ) self.presenter.dead_time_presenter.handle_run_selector_changed = mock.Mock( ) self.presenter.dead_time_presenter.handle_pre_process_and_counts_calculated = mock.Mock( ) self.presenter.background_presenter.initialize_model_options = mock.Mock( ) self.presenter.background_presenter.handle_instrument_changed = mock.Mock( ) self.presenter.background_presenter.handle_runs_loaded = mock.Mock() self.presenter.background_presenter.handle_run_selector_changed = mock.Mock( ) self.presenter.background_presenter.handle_groups_changed = mock.Mock() self.presenter.background_presenter.handle_pre_process_and_counts_calculated = mock.Mock( ) self.presenter.background_presenter.handle_background_corrections_for_all_finished = mock.Mock( ) self.presenter._handle_selected_table_is_invalid = mock.Mock() self.presenter._notify_perform_dead_time_corrections = mock.Mock() self.presenter._perform_asymmetry_pairs_and_diffs_calculation = mock.Mock( ) # Mock the correction result property self.presenter.thread_model_wrapper = mock.Mock(spec=ThreadModel) self.mock_presenter_correction_results = mock.PropertyMock( return_value=(self.run_strings, self.groups)) type(self.presenter.thread_model_wrapper ).result = self.mock_presenter_correction_results
def setUp(self): self.filepath = FileFinder.findRuns('EMU00019489.nxs')[0] self.load_result, self.run_number, self.filename, _ = load_workspace_from_filename( self.filepath) self.context = setup_context() self.context.gui_context.update({'RebinType': 'None'}) self.model = CorrectionsModel(self.context) self.runs = [[84447], [84448], [84449]] self.coadd_runs = [[84447, 84448, 84449]] self.context.data_context.instrument = 'EMU' self.context.data_context._loaded_data.add_data( workspace=self.load_result, run=[self.run_number], filename=self.filename, instrument='EMU') self.context.data_context.current_runs = [[self.run_number]] self.context.data_context.update_current_data() self.context.group_pair_context.reset_group_and_pairs_to_default( self.load_result['OutputWorkspace'][0].workspace, 'EMU', '', 1)
def setUp(self): self.context = setup_context(True) # create widget self.widget = transform_widget.TransformWidget(self.context, FFTWidget, MaxEntWidget) self.widget._maxent._presenter = mock.MagicMock( ) #create_autospec(maxent_presenter_new.MaxEntPresenter, spec_set=True) self.widget._fft._presenter = mock.MagicMock( ) #create_autospec(fft_presenter_new.FFTPresenter, spec_Set=True) # create the view self.view = mock.create_autospec(transform_view.TransformView, spec_set=False) self.view.getView = mock.Mock() self.view.getMethods = mock.Mock(return_value=["FFT", "MaxEnt"]) self.view.hideAll = mock.Mock() self.view.showMethod = mock.Mock() self.view.selection = mock.create_autospec( transform_selection_view.TransformSelectionView, spec_set=True) self.view.selection.changeMethodSignal = mock.Mock() # set the mocked view to the widget self.widget.mockWidget(self.view)
def setUp(self): context = setup_context() context._do_rebin = mock.Mock(return_value=False) self.corrections_model = CorrectionsModel(context) self.model = BackgroundCorrectionsModel(self.corrections_model, context) self.model.clear_background_corrections_data() self.runs = ["84447", "84447", "84447", "84447"] self.groups = ["fwd", "bwd", "top", "bottom"] self.use_raws = [DEFAULT_USE_RAW] * len(self.groups) self.start_xs = [15.0] * 4 self.end_xs = [30.0] * 4 self.a0s = [0.0] * 4 self.a0_errors = [0.0] * 4 self.statuses = ["No background correction"] * 4 self.fitted_a0 = 1.234 self.fitted_a0_error = 0.005 self.fitted_function = self._setup_fitted_function() self.fit_status = "success" self.chi_squared = 2.2
def setUp(self): context = setup_context() self.model = TFAsymmetryFittingModel(context, context.fitting_context) self.dataset_names = [ "EMU20884; Group; fwd; Asymmetry", "EMU20884; Group; top; Asymmetry" ] self.tf_non_compliant_dataset_names = [ "EMU20884; Group; fwd; Asymmetry", "EMU20884; Pair Asym; long; Asymmetry" ] self.fit_function = FunctionFactory.createFunction("FlatBackground") self.single_fit_functions = [ self.fit_function.clone(), self.fit_function.clone() ] self.simultaneous_fit_function = FunctionFactory.createInitializedMultiDomainFunction( str(self.fit_function), len(self.dataset_names)) self.tf_single_function = create_tf_asymmetry_function( self.fit_function.clone()) self.tf_single_fit_functions = [ self.tf_single_function.clone(), self.tf_single_function.clone() ] self.tf_simultaneous_fit_function = FunctionFactory.createInitializedMultiDomainFunction( str(self.tf_single_function), len(self.dataset_names)) CreateSampleWorkspace(Function="One Peak", XMin=0.0, XMax=15.0, BinWidth=0.1, OutputWorkspace=self.dataset_names[0]) CreateSampleWorkspace(Function="One Peak", XMin=0.0, XMax=15.0, BinWidth=0.1, OutputWorkspace=self.dataset_names[1])
def setUp(self): self.loaded_data = MuonLoadData() self.context = setup_context(False) self.data_context = self.context.data_context self.gui_context = self.context.gui_context self.group_context = self.context.group_pair_context self.model = GroupingTabModel(context=self.context) self.grouping_table_view = GroupingTableView() self.grouping_table_widget = GroupingTablePresenter( self.grouping_table_view, self.model) self.pairing_table_view = PairingTableView() self.pairing_table_widget = PairingTablePresenter( self.pairing_table_view, self.model) self.grouping_table_view.warning_popup = mock.MagicMock() self.pairing_table_view.warning_popup = mock.MagicMock() self.add_three_groups() self.add_two_pairs() self.view = GroupingTabView(self.grouping_table_view, self.pairing_table_view) self.presenter = GroupingTabPresenter(self.view, self.model, self.grouping_table_widget, self.pairing_table_widget) self.presenter.create_update_thread = mock.MagicMock( return_value=mock.MagicMock()) self.view.display_warning_box = mock.MagicMock() self.grouping_table_view.warning_popup = mock.MagicMock() self.pairing_table_view.warning_popup = mock.MagicMock()
def setUp(self): context = setup_context() self.corrections_model = CorrectionsModel(context) self.model = DeadTimeCorrectionsModel(self.corrections_model, context.data_context, context.corrections_context) self.runs = [[84447], [84448], [84449]] self.coadd_runs = [[84447, 84448, 84449]]
def setUp(self): self.model = FittingTabModel(setup_context())
def setUp(self): self.model = FittingTabModel(setup_context()) self.model.context.ads_observer.unsubscribe()
def setUp(self): self.model = BasePaneModel(context=setup_context(False))
def setUpClass(cls): cls.context = setup_context()
def setUp(self): self.context = setup_context(False) self.context.data_context.instrument = "MUSR" self.model = RawPaneModel(self.context) self.name = self.model.name
def setUp(self): self.context = setup_context() self.view = mock.MagicMock() self.model = mock.MagicMock() self.presenter = SeqFittingTabPresenter(self.view, self.model, self.context)