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 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)
