class WorkspaceWidgetTest(unittest.TestCase, QtWidgetFinder):
def setUp(self):
self.ws_widget = WorkspaceWidget(QMainWindow())
mat_ws = CreateSampleWorkspace()
table_ws = CreateEmptyTableWorkspace()
group_ws = GroupWorkspaces([mat_ws, table_ws])
single_val_ws = CreateSingleValuedWorkspace(5, 6)
self.w_spaces = [mat_ws, table_ws, group_ws, single_val_ws]
self.ws_names = ['MatWS', 'TableWS', 'GroupWS', 'SingleValWS']
for ws_name, ws in zip(self.ws_names, self.w_spaces):
self.ws_widget._ads.add(ws_name, ws)
def tearDown(self):
self.ws_widget._ads.clear()
def test_algorithm_history_window_opens_with_workspace(self):
with mock.patch(ALGORITHM_HISTORY_WINDOW + '.show', lambda x: None):
self.ws_widget._do_show_algorithm_history([self.ws_names[0]])
self.assert_widget_type_exists(ALGORITHM_HISTORY_WINDOW_TYPE)
def test_algorithm_history_window_doesnt_open_with_workspace_group(self):
with mock.patch(ALGORITHM_HISTORY_WINDOW + '.show', lambda x: None):
self.ws_widget._do_show_algorithm_history([self.ws_names[2]])
self.assert_widget_type_doesnt_exist(ALGORITHM_HISTORY_WINDOW_TYPE)
def test_algorithm_history_window_opens_multiple(self):
with mock.patch(ALGORITHM_HISTORY_WINDOW + '.show', lambda x: None):
self.ws_widget._do_show_algorithm_history(self.ws_names)
self.assert_number_of_widgets_matching(ALGORITHM_HISTORY_WINDOW_TYPE,
3)
def test_detector_table_shows_with_workspace(self):
with mock.patch(MATRIXWORKSPACE_DISPLAY + '.show_view',
lambda x: None):
self.ws_widget._do_show_detectors([self.ws_names[0]])
self.assert_widget_type_exists(MATRIXWORKSPACE_DISPLAY_TYPE)
@mock.patch('workbench.plugins.workspacewidget.plot', autospec=True)
def test_plot_with_plot_bin(self, mock_plot):
self.ws_widget._do_plot_bin([self.ws_names[0]], False, False)
mock_plot.assert_called_once_with(
mock.ANY,
errors=False,
overplot=False,
wksp_indices=[0],
plot_kwargs={'axis': MantidAxType.BIN})
@mock.patch('workbench.plugins.workspacewidget.plot_from_names',
autospec=True)
def test_plot_with_plot_spectrum(self, mock_plot_from_names):
self.ws_widget._do_plot_spectrum([self.ws_names[0]], False, False)
mock_plot_from_names.assert_called_once_with([self.ws_names[0]],
False,
False,
advanced=False)
@mock.patch('workbench.plugins.workspacewidget.pcolormesh', autospec=True)
def test_plot_with_plot_colorfill(self, mock_plot_colorfill):
self.ws_widget._do_plot_colorfill([self.ws_names[0]])
mock_plot_colorfill.assert_called_once_with(mock.ANY)
@mock.patch('workbench.plugins.workspacewidget.plot_from_names',
autospec=True)
def test_plot_with_plot_advanced(self, mock_plot_from_names):
self.ws_widget._do_plot_spectrum([self.ws_names[0]],
False,
False,
advanced=True)
mock_plot_from_names.assert_called_once_with([self.ws_names[0]],
False,
False,
advanced=True)
@mock.patch('mantidqt.plotting.functions.plot_contour', autospec=True)
def test_plot_with_plot_contour(self, mock_plot_contour):
self.ws_widget._do_plot_3D([self.ws_names[0]], plot_type='contour')
mock_plot_contour.assert_called_once_with([self.ws_names[0]])
@mock.patch('mantidqt.plotting.functions.plot_surface', autospec=True)
def test_plot_with_plot_surface(self, mock_plot_surface):
self.ws_widget._do_plot_3D([self.ws_names[0]], plot_type='surface')
mock_plot_surface.assert_called_once_with([self.ws_names[0]])
@mock.patch('mantidqt.plotting.functions.plot_wireframe', autospec=True)
def test_plot_with_plot_wireframe(self, mock_plot_wireframe):
self.ws_widget._do_plot_3D([self.ws_names[0]], plot_type='wireframe')
mock_plot_wireframe.assert_called_once_with([self.ws_names[0]])
def test_double_click_with_single_value_ws_shows_data(self):
with mock.patch(MATRIXWORKSPACE_DISPLAY + '.show_view',
lambda x: None):
self.ws_widget._action_double_click_workspace(self.ws_names[3])
self.assert_widget_type_exists(MATRIXWORKSPACE_DISPLAY_TYPE)
def test_empty_workspaces(self):
self.ws_widget._ads.clear()
self.assertEqual(self.ws_widget.empty_of_workspaces(), True)
CreateSampleWorkspace(OutputWorkspace="ws")
self.assertEqual(self.ws_widget.empty_of_workspaces(), False)
class WorkspaceWidgetTest(unittest.TestCase, QtWidgetFinder):
def setUp(self):
self.ws_widget = WorkspaceWidget(QMainWindow())
mat_ws = CreateSampleWorkspace()
table_ws = CreateEmptyTableWorkspace()
group_ws = GroupWorkspaces([mat_ws, table_ws])
single_val_ws = CreateSingleValuedWorkspace(5, 6)
self.w_spaces = [mat_ws, table_ws, group_ws, single_val_ws]
self.ws_names = ['MatWS', 'TableWS', 'GroupWS', 'SingleValWS']
# create md workspace
md_ws = CreateMDHistoWorkspace(SignalInput='1,2,3,4,2,1',
ErrorInput='1,1,1,1,1,1',
Dimensionality=3,
Extents='-1,1,-1,1,0.5,6.5',
NumberOfBins='1,1,6',
Names='x,y,|Q|',
Units='mm,km,AA^-1',
OutputWorkspace='MDHistoWS1D')
# self.w_spaces = [mat_ws, table_ws, group_ws, single_val_ws, md_ws]
# self.ws_names = ['MatWS', 'TableWS', 'GroupWS', 'SingleValWS', 'MDHistoWS1D']
for ws_name, ws in zip(self.ws_names, self.w_spaces):
self.ws_widget._ads.add(ws_name, ws)
self.ws_names.append(md_ws.name())
self.w_spaces.append(md_ws)
def tearDown(self):
self.ws_widget._ads.clear()
def test_algorithm_history_window_opens_with_workspace(self):
with mock.patch(ALGORITHM_HISTORY_WINDOW + '.show', lambda x: None):
self.ws_widget._do_show_algorithm_history([self.ws_names[0]])
self.assert_widget_type_exists(ALGORITHM_HISTORY_WINDOW_TYPE)
def test_algorithm_history_window_doesnt_open_with_workspace_group(self):
with mock.patch(ALGORITHM_HISTORY_WINDOW + '.show', lambda x: None):
self.ws_widget._do_show_algorithm_history([self.ws_names[2]])
self.assert_widget_type_doesnt_exist(ALGORITHM_HISTORY_WINDOW_TYPE)
def test_algorithm_history_window_opens_multiple(self):
"""
There are 5 objects in ADS. But 1 of them is WorkspaceGroup
This sets total number of history windows to 4.
:return:
"""
with mock.patch(ALGORITHM_HISTORY_WINDOW + '.show', lambda x: None):
self.ws_widget._do_show_algorithm_history(self.ws_names)
self.assert_number_of_widgets_matching(ALGORITHM_HISTORY_WINDOW_TYPE,
4)
def test_detector_table_shows_with_workspace(self):
with mock.patch(MATRIXWORKSPACE_DISPLAY + '.show_view',
lambda x: None):
self.ws_widget._do_show_detectors([self.ws_names[0]])
self.assert_widget_type_exists(MATRIXWORKSPACE_DISPLAY_TYPE)
@mock.patch('workbench.plugins.workspacewidget.plot', autospec=True)
def test_plot_with_plot_bin(self, mock_plot):
self.ws_widget._do_plot_bin([self.ws_names[0]], False, False)
mock_plot.assert_called_once_with(
mock.ANY,
errors=False,
overplot=False,
wksp_indices=[0],
plot_kwargs={'axis': MantidAxType.BIN})
@mock.patch('workbench.plugins.workspacewidget.plot_from_names',
autospec=True)
def test_plot_with_plot_spectrum(self, mock_plot_from_names):
self.ws_widget._do_plot_spectrum([self.ws_names[0]], False, False)
mock_plot_from_names.assert_called_once_with([self.ws_names[0]],
False,
False,
advanced=False)
@mock.patch('workbench.plugins.workspacewidget.plot_md_ws_from_names',
autospec=True)
def test_plot_with_1d_mdhistoworkspace(self, mock_plot_md_from_names):
self.ws_widget._do_plot_1d_md([self.ws_names[4]], False, False)
mock_plot_md_from_names.assert_called_once_with([self.ws_names[4]],
False, False)
@mock.patch('workbench.plugins.workspacewidget.pcolormesh', autospec=True)
def test_plot_with_plot_colorfill(self, mock_plot_colorfill):
self.ws_widget._do_plot_colorfill([self.ws_names[0]])
mock_plot_colorfill.assert_called_once_with(mock.ANY)
@mock.patch('workbench.plugins.workspacewidget.plot_from_names',
autospec=True)
def test_plot_with_plot_advanced(self, mock_plot_from_names):
self.ws_widget._do_plot_spectrum([self.ws_names[0]],
False,
False,
advanced=True)
mock_plot_from_names.assert_called_once_with([self.ws_names[0]],
False,
False,
advanced=True)
@mock.patch('mantidqt.plotting.functions.plot_contour', autospec=True)
def test_plot_with_plot_contour(self, mock_plot_contour):
self.ws_widget._do_plot_3D([self.ws_names[0]], plot_type='contour')
mock_plot_contour.assert_called_once_with([self.ws_names[0]])
@mock.patch('mantidqt.plotting.functions.plot_surface', autospec=True)
def test_plot_with_plot_surface(self, mock_plot_surface):
self.ws_widget._do_plot_3D([self.ws_names[0]], plot_type='surface')
mock_plot_surface.assert_called_once_with([self.ws_names[0]])
@mock.patch('mantidqt.plotting.functions.plot_wireframe', autospec=True)
def test_plot_with_plot_wireframe(self, mock_plot_wireframe):
self.ws_widget._do_plot_3D([self.ws_names[0]], plot_type='wireframe')
mock_plot_wireframe.assert_called_once_with([self.ws_names[0]])
def test_double_click_with_single_value_ws_shows_data(self):
with mock.patch(MATRIXWORKSPACE_DISPLAY + '.show_view',
lambda x: None):
self.ws_widget._action_double_click_workspace(self.ws_names[3])
self.assert_widget_type_exists(MATRIXWORKSPACE_DISPLAY_TYPE)
def test_empty_workspaces(self):
self.ws_widget._ads.clear()
self.assertEqual(self.ws_widget.empty_of_workspaces(), True)
CreateSampleWorkspace(OutputWorkspace="ws")
self.assertEqual(self.ws_widget.empty_of_workspaces(), False)