def test_setup_table(self): ws = MockWorkspace() view = MockMatrixWorkspaceDisplayView() container = Mock(spec=StatusBarView) container.status_bar = Mock(spec=QStatusBar) MatrixWorkspaceDisplay(ws, view=view, container=container) self.assertEqual(3, view.set_context_menu_actions.call_count) self.assertEqual(1, view.set_model.call_count)
def wrapper(self, *args): ws = MockWorkspace() view = MockMatrixWorkspaceDisplayView() mock_observer = Mock() container = Mock(spec=StatusBarView) container.status_bar = Mock(spec=QStatusBar) presenter = MatrixWorkspaceDisplay(ws, view=view, ads_observer=mock_observer, container=container) return func(self, ws, view, presenter, *args)
def common_setup_action_plot(self, table_has_selection=True): mock_ws = MockWorkspace() mock_view = MockMatrixWorkspaceDisplayView() mock_plotter = Mock() container = Mock(spec=StatusBarView) container.status_bar = Mock(spec=QStatusBar) presenter = MatrixWorkspaceDisplay(mock_ws, plot=mock_plotter, view=mock_view, container=container) # monkey-patch the spectrum plot label to count the number of calls presenter.model.get_spectrum_plot_label = Mock() presenter.model.get_bin_plot_label = Mock() mock_table = MockQTableView() # configure the mock return values mock_table.mock_selection_model.hasSelection = Mock(return_value=table_has_selection) return mock_plotter, mock_table, mock_view, presenter
def wrapper(self, *args): ws = MockWorkspace() view = Mock(spec=TableWorkspaceDisplayView) container = Mock(spec=StatusBarView) container.status_bar = Mock(spec=QStatusBar) if add_selection_model: mock_selection_model = MockQSelectionModel(has_selection=True) mock_selection_model.selectedRows = Mock( return_value=[MockQModelIndex(1, 1), MockQModelIndex(2, 2), MockQModelIndex(3, 3)]) mock_selection_model.selectedColumns = Mock( return_value=[MockQModelIndex(1, 1), MockQModelIndex(2, 2), MockQModelIndex(3, 3)]) view.mock_selection_model = mock_selection_model view.selectionModel.return_value = mock_selection_model twd = TableWorkspaceDisplay(ws, view=view, container=container) if add_plot: twd.plot = MockPlotLib() return func(self, ws, view, twd, *args)
def common_setup_action_plot(self, table_has_selection=True): mock_ws = MockWorkspace() mock_view = MockMatrixWorkspaceDisplayView() mock_plotter = Mock() container = Mock(spec=StatusBarView) container.status_bar = Mock(spec=QStatusBar) presenter = MatrixWorkspaceDisplay(mock_ws, plot=mock_plotter, view=mock_view, container=container) # monkey-patch the spectrum plot label to count the number of calls presenter.model.get_spectrum_plot_label = Mock() presenter.model.get_bin_plot_label = Mock() mock_table = MockQTableView() # configure the mock return values mock_table.mock_selection_model.hasSelection = Mock( return_value=table_has_selection) return mock_plotter, mock_table, mock_view, presenter
def wrapper(self, *args): ws = MockWorkspace() view = Mock(spec=TableWorkspaceDisplayView) container = Mock(spec=StatusBarView) container.status_bar = Mock(spec=QStatusBar) if add_selection_model: mock_selection_model = MockQSelectionModel(has_selection=True) mock_selection_model.selectedRows = Mock(return_value=[ MockQModelIndex(1, 1), MockQModelIndex(2, 2), MockQModelIndex(3, 3) ]) mock_selection_model.selectedColumns = Mock(return_value=[ MockQModelIndex(1, 1), MockQModelIndex(2, 2), MockQModelIndex(3, 3) ]) view.mock_selection_model = mock_selection_model view.selectionModel.return_value = mock_selection_model twd = TableWorkspaceDisplay(ws, view=view, container=container) if add_plot: twd.plot = MockPlotLib() return func(self, ws, view, twd, *args)