def test_add_observers_to_feature(self, observer):
group_observer = mock.Mock()
observer.return_value = group_observer
add = AddGroupingWorkspaces(self.GUI, {})
add.add_observers_to_feature(self.GUI)
self.GUI.corrections_tab.corrections_tab_presenter.asymmetry_pair_and_diff_calculations_finished_notifier.\
add_subscriber.assert_called_once_with(group_observer)
self.GUI.phase_tab.phase_table_presenter.calculation_finished_notifier.add_subscriber.\
assert_called_once_with(group_observer)
self.GUI.phase_tab.phase_table_presenter.phase_table_calculation_complete_notifier.add_subscriber.\
assert_called_once_with(group_observer)
def test_add_observers_to_feature_with_frequency(self, observer):
group_observer = mock.Mock()
observer.return_value = group_observer
self.GUI.transform = mock.MagicMock(autospec=TransformWidget)
add = AddGroupingWorkspaces(self.GUI, {})
add.add_observers_to_feature(self.GUI)
self.GUI.corrections_tab.corrections_tab_presenter.asymmetry_pair_and_diff_calculations_finished_notifier.\
add_subscriber.assert_called_once_with(group_observer)
self.GUI.phase_tab.phase_table_presenter.calculation_finished_notifier.add_subscriber.\
assert_called_once_with(group_observer)
self.GUI.phase_tab.phase_table_presenter.phase_table_calculation_complete_notifier.add_subscriber.\
assert_called_once_with(group_observer)
self.GUI.transform.new_data_observer.assert_called_once_with(group_observer)
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 test_get_features_success(self, observer):
test = {}
AddGroupingWorkspaces(self.GUI, test)
observer.assert_called_once_with(self.GUI.context.do_grouping)