def setup_context(freq=False): loaded_data = MuonLoadData() loaded_data.get_main_field_direction = mock.MagicMock(return_value='transverse') data_context = MuonDataContext(load_data=loaded_data) gui_context = MuonGuiContext() group_context = MuonGroupPairContext(data_context.check_group_contains_valid_detectors) corrections_context = CorrectionsContext(loaded_data) phase_table_context = PhaseTableContext() freq_context = FrequencyContext() plot_panes_context = PlotPanesContext() if freq: return FrequencyDomainAnalysisContext(muon_data_context=data_context, muon_group_context=group_context, muon_gui_context=gui_context, muon_phase_context=phase_table_context, corrections_context=corrections_context, fitting_context=BasicFittingContext(allow_double_pulse_fitting=True), frequency_context=freq_context, plot_panes_context=plot_panes_context) else: return DataAnalysisContext(muon_data_context=data_context, muon_group_context=group_context, muon_gui_context=gui_context, corrections_context=corrections_context, muon_phase_context=phase_table_context, fitting_context=TFAsymmetryFittingContext(allow_double_pulse_fitting=True), results_context=ResultsContext(), model_fitting_context=ModelFittingContext(), plot_panes_context=plot_panes_context)
def setup_context_for_tests(parent_object): parent_object.loaded_data = MuonLoadData() parent_object.loaded_data.get_main_field_direction = mock.MagicMock(return_value='transverse') parent_object.data_context = MuonDataContext(load_data=parent_object.loaded_data) parent_object.gui_context = MuonGuiContext() parent_object.group_context = MuonGroupPairContext(parent_object.data_context.check_group_contains_valid_detectors) parent_object.corrections_context = CorrectionsContext(parent_object.loaded_data) parent_object.phase_table_context = PhaseTableContext() parent_object.fitting_context = TFAsymmetryFittingContext(allow_double_pulse_fitting=True) parent_object.results_context = ResultsContext() parent_object.plot_panes_context = PlotPanesContext() parent_object.model_fitting_context = ModelFittingContext() parent_object.context = DataAnalysisContext(muon_data_context=parent_object.data_context, muon_group_context=parent_object.group_context, muon_gui_context=parent_object.gui_context, muon_phase_context=parent_object.phase_table_context, corrections_context=parent_object.corrections_context, fitting_context=parent_object.fitting_context, results_context=parent_object.results_context, model_fitting_context=parent_object.model_fitting_context, plot_panes_context=parent_object.plot_panes_context)
def __init__(self, parent=None, window_flags=None): super(FrequencyAnalysisGui, self).__init__(parent) if window_flags: self.setWindowFlags(window_flags) self.setAttribute(QtCore.Qt.WA_DeleteOnClose) self.setFocusPolicy(QtCore.Qt.StrongFocus) try: check_facility() except AttributeError as error: self.warning_popup(error.args[0]) # load the feature flags feature_dict = load_features() # initialise the data storing classes of the interface self.loaded_data = MuonLoadData() self.data_context = MuonDataContext('Frequency Domain Data', self.loaded_data) self.gui_context = MuonGuiContext() self.plot_panes_context = PlotPanesContext() self.group_pair_context = MuonGroupPairContext( self.data_context.check_group_contains_valid_detectors) self.corrections_context = CorrectionsContext(self.loaded_data) self.phase_context = PhaseTableContext() self.fitting_context = BasicFittingContext( allow_double_pulse_fitting=True) self.results_context = ResultsContext() self.model_fitting_context = ModelFittingContext( allow_double_pulse_fitting=False) self.frequency_context = FrequencyContext() self.context = FrequencyDomainAnalysisContext( muon_data_context=self.data_context, muon_gui_context=self.gui_context, muon_group_context=self.group_pair_context, corrections_context=self.corrections_context, muon_phase_context=self.phase_context, plot_panes_context=self.plot_panes_context, fitting_context=self.fitting_context, results_context=self.results_context, model_fitting_context=self.model_fitting_context, frequency_context=self.frequency_context) # create the dockable widget self.plot_widget = FrequencyAnalysisPlotWidget(self.context, parent=self) self.dockable_plot_widget_window = PlottingDockWidget( parent=self, plotting_widget=self.plot_widget.view) self.dockable_plot_widget_window.setMinimumWidth(575) # Add dock widget to main Muon analysis window self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.dockable_plot_widget_window) # Need this line to stop the bug where the dock window snaps back to its original size after resizing. # 0 argument is arbitrary and has no effect on fit widget size # This is a qt bug reported at (https://bugreports.qt.io/browse/QTBUG-65592) if QT_VERSION >= LooseVersion("5.6"): self.resizeDocks({self.dockable_plot_widget_window}, {1}, QtCore.Qt.Horizontal) # construct all the widgets. self.load_widget = LoadWidget(self.loaded_data, self.context, self) self.grouping_tab_widget = GroupingTabWidget(self.context, parent) self.corrections_tab = CorrectionsTabWidget(self.context, self) self.home_tab = HomeTabWidget(self.context, self) self.phase_tab = PhaseTabWidget(self.context, self) self.transform = TransformWidget(self.context, FFTWidget, MaxEntWidget, parent=self) self.fitting_tab = FittingTabWidget(self.context, self) self.seq_fitting_tab = SeqFittingTabWidget( self.context, self.fitting_tab.fitting_tab_model, self) self.results_tab = ResultsTabWidget(self.context.fitting_context, self.context, self) self.add_model_analysis = AddModelAnalysis(self, feature_dict) self.add_raw_plots = AddRawPlots(self, feature_dict) self.add_fitting = AddFitting(self, feature_dict) setup_group_ws = AddGroupingWorkspaces(self, feature_dict) self.setup_tabs() self.plot_widget.insert_plot_panes() self.help_widget = HelpWidget(self.context.window_title) central_widget = QtWidgets.QWidget() vertical_layout = QtWidgets.QVBoxLayout() vertical_layout.addWidget(self.load_widget.load_widget_view) vertical_layout.addWidget(self.tabs) vertical_layout.addWidget(self.help_widget.view) central_widget.setLayout(vertical_layout) central_widget.setSizePolicy( QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Maximum, QtWidgets.QSizePolicy.Maximum)) self.disable_notifier = GenericObservable() self.disable_observer = GenericObserver( self.disable_notifier.notify_subscribers) self.enable_notifier = GenericObservable() self.enable_observer = GenericObserver( self.enable_notifier.notify_subscribers) self.setup_disable_notifier() self.setup_enable_notifier() self.setCentralWidget(central_widget) self.setWindowTitle(self.context.window_title) self.setup_load_observers() self.setup_gui_variable_observers() self.setup_grouping_changed_observers() self.setup_corrections_changed_observers() self.setup_instrument_changed_notifier() self.setup_group_calculation_enable_notifier() self.setup_group_calculation_disabler_notifier() self.setup_on_load_enabler() self.setup_on_load_disabler() self.setup_phase_quad_changed_notifier() self.setup_phase_table_changed_notifier() self.setup_fitting_notifier() self.setup_counts_calculation_finished_notifier() self.setup_asymmetry_pair_and_diff_calculations_finished_notifier() self.setup_transform() self.context.data_context.message_notifier.add_subscriber( self.grouping_tab_widget.group_tab_presenter.message_observer) self.add_model_analysis.add_observers_to_feature(self) self.add_model_analysis.set_feature_observables(self) self.add_raw_plots.add_observers_to_feature(self) setup_group_ws.add_observers_to_feature(self)
def setUp(self): self.muon_data = MuonLoadData() self.corrections_context = CorrectionsContext(self.muon_data)
class CorrectionsContextTest(unittest.TestCase): def setUp(self): self.muon_data = MuonLoadData() self.corrections_context = CorrectionsContext(self.muon_data) def test_that_the_context_has_been_instantiated_with_the_expected_context_data( self): self.assertEqual(self.corrections_context.current_run_string, None) self.assertEqual(self.corrections_context.dead_time_source, "FromFile") self.assertEqual( self.corrections_context.dead_time_table_name_from_ads, None) self.assertEqual(self.corrections_context.background_corrections_mode, "None") self.assertEqual(self.corrections_context.selected_function, "Flat Background + Exp Decay") self.assertEqual(self.corrections_context.selected_group, "All") self.assertEqual(self.corrections_context.show_all_runs, False) def test_that_the_current_run_string_can_be_set_as_expected(self): run_string = "62260" self.corrections_context.current_run_string = run_string self.assertEqual(self.corrections_context.current_run_string, run_string) def test_that_the_dead_time_source_can_be_set_as_expected(self): source = "FromADS" self.corrections_context.dead_time_source = source self.assertEqual(self.corrections_context.dead_time_source, source) def test_that_the_dead_time_table_name_from_ads_can_be_set_as_expected( self): table_name = "MUSR62260 dead time table" self.corrections_context.dead_time_table_name_from_ads = table_name self.assertEqual( self.corrections_context.dead_time_table_name_from_ads, table_name) def test_that_the_dead_time_table_name_from_file_can_be_set_as_expected( self): table_name = "MUSR62260 dead time table" self.corrections_context.dead_time_table_name_from_file = table_name self.assertEqual( self.corrections_context.dead_time_table_name_from_file, table_name) def test_that_current_dead_time_table_name_returns_the_expected_table_when_the_source_is_from_file( self): table_from_file = "MUSR62260 dead time table" table_from_ads = "MUSR62265 dead time table" self.corrections_context.get_default_dead_time_table_name_for_run = mock.Mock( return_value=table_from_file) self.corrections_context.dead_time_source = "FromFile" self.corrections_context.dead_time_table_name_from_ads = table_from_ads self.assertEqual( self.corrections_context.current_dead_time_table_name_for_run( "MUSR", [62260]), table_from_file) def test_that_current_dead_time_table_name_returns_the_expected_table_when_the_source_is_from_ads( self): table_from_ads = "MUSR62265 dead time table" self.corrections_context.dead_time_source = "FromADS" self.corrections_context.dead_time_table_name_from_ads = table_from_ads self.assertEqual( self.corrections_context.current_dead_time_table_name_for_run( "MUSR", [62265]), table_from_ads) def test_that_current_dead_time_table_name_returns_none_when_the_source_is_from_none( self): table_from_ads = "MUSR62265 dead time table" self.corrections_context.dead_time_source = None self.corrections_context.dead_time_table_name_from_ads = table_from_ads self.assertEqual( self.corrections_context.current_dead_time_table_name_for_run( "MUSR", [62265]), None) def test_that_the_background_corrections_mode_can_be_set_as_expected(self): background_corrections_mode = "Auto" self.corrections_context.background_corrections_mode = background_corrections_mode self.assertEqual(self.corrections_context.background_corrections_mode, background_corrections_mode) def test_that_the_selected_function_can_be_set_as_expected(self): selected_function = "Flat Background + Exp Decay" self.corrections_context.selected_function = selected_function self.assertEqual(self.corrections_context.selected_function, selected_function) def test_that_the_selected_group_can_be_set_as_expected(self): selected_group = "fwd" self.corrections_context.selected_group = selected_group self.assertEqual(self.corrections_context.selected_group, selected_group) def test_that_show_all_runs_can_be_set_as_expected(self): show_all_runs = True self.corrections_context.show_all_runs = show_all_runs self.assertEqual(self.corrections_context.show_all_runs, show_all_runs) def test_that_the_background_correction_data_can_be_set_as_expected(self): run_group = tuple(["84447", "fwd"]) start_x, end_x = 15.0, 30.0 self.corrections_context.background_correction_data[ run_group] = BackgroundCorrectionData(True, 5, start_x, end_x) self.assertTrue( run_group in self.corrections_context.background_correction_data) self.assertEqual( self.corrections_context.background_correction_data[run_group]. use_raw, True) self.assertEqual( self.corrections_context.background_correction_data[run_group]. rebin_fixed_step, 5) self.assertEqual( self.corrections_context.background_correction_data[run_group]. start_x, start_x) self.assertEqual( self.corrections_context.background_correction_data[run_group]. end_x, end_x)