def test_make_fig_makes_the_right_calls(self, pass_func):
ws = Workspace2D()
ADS.add("ws", ws)
plot_dict = {
"label":
"plot",
"creationArguments": [[{
"workspaces": "ws",
"wkspIndex": 0
}, {
"function": "axhline",
"args": [10, 0, 1],
"kwargs": {}
}, {
"function": "axvline",
"args": [],
"kwargs": {
"x": 0,
"ymin": 0,
"ymax": 1
}
}]]
}
self.plots_loader.workspace_plot_func = mock.MagicMock()
self.plots_loader.plot_func = mock.MagicMock()
self.plots_loader.restore_figure_data = mock.MagicMock()
self.plots_loader.make_fig(plot_dict)
self.assertEqual(self.plots_loader.workspace_plot_func.call_count, 1)
self.assertEqual(self.plots_loader.plot_func.call_count, 2)
self.assertEqual(self.plots_loader.restore_figure_data.call_count, 1)
def test_addOrReplace_replaces_workspace_with_existing_name(self):
data = [1.0,2.0,3.0]
alg = run_algorithm('CreateWorkspace',DataX=data,DataY=data,NSpec=1,UnitX='Wavelength', child=True)
name = "testws"
ws = alg.getProperty("OutputWorkspace").value
AnalysisDataService.add(name, ws)
len_before = len(AnalysisDataService)
AnalysisDataService.addOrReplace(name, ws)
len_after = len(AnalysisDataService)
self.assertEqual(len_after, len_before)
def test_addOrReplace_replaces_workspace_with_existing_name(self):
data = [1.0,2.0,3.0]
alg = run_algorithm('CreateWorkspace',DataX=data,DataY=data,NSpec=1,UnitX='Wavelength', child=True)
name = "testws"
ws = alg.getProperty("OutputWorkspace").value
AnalysisDataService.add(name, ws)
len_before = len(AnalysisDataService)
AnalysisDataService.addOrReplace(name, ws)
len_after = len(AnalysisDataService)
self.assertEquals(len_after, len_before)
def test_make_fig_makes_the_right_calls(self, pass_func):
ws = Workspace2D()
ADS.add("ws", ws)
plot_dict = {"label": "plot", "creationArguments": [[{"workspaces": "ws", "wkspIndex": 0}, {}, {}]]}
self.plots_loader.plot_func = mock.MagicMock()
self.plots_loader.restore_figure_data = mock.MagicMock()
self.plots_loader.make_fig(plot_dict)
self.assertEqual(self.plots_loader.plot_func.call_count, 1)
self.assertEqual(self.plots_loader.restore_figure_data.call_count, 1)
def test_that_a_histogram_workspace_is_returned_as_a_MatrixWorkspace_from_ADS(self):
wsname = "MatrixWorkspaceTest_ADS"
AnalysisDataService.add(wsname, self._test_ws)
value = AnalysisDataService[wsname]
self.assertTrue(isinstance(value, Workspace))
# Have got a MatrixWorkspace back and not just the generic interface
self.assertTrue(isinstance(value, MatrixWorkspace))
mem = value.getMemorySize()
self.assertTrue((mem > 0))
AnalysisDataService.remove(wsname)
def test_that_a_histogram_workspace_is_returned_as_a_MatrixWorkspace_from_a_property(self):
wsname = "MatrixWorkspaceTest_Property"
AnalysisDataService.add(wsname, self._test_ws)
alg = create_algorithm("Rebin", InputWorkspace=wsname)
propValue = alg.getProperty("InputWorkspace").value
# Is Workspace in the hierarchy of the value
self.assertTrue(isinstance(propValue, Workspace))
# Have got a MatrixWorkspace back and not just the generic interface
self.assertTrue(isinstance(propValue, MatrixWorkspace))
mem = propValue.getMemorySize()
self.assertTrue((mem > 0))
AnalysisDataService.remove(wsname)
def test_MaskWorkspace_Is_Retrievable(self):
dummy_ws = WorkspaceCreationHelper.create2DWorkspaceWithFullInstrument(2, 102, False) # no monitors
ws_name = "dummy"
AnalysisDataService.add(ws_name, dummy_ws)
run_algorithm('MaskDetectors', Workspace=ws_name, WorkspaceIndexList=1)
mask_name = 'mask_ws'
run_algorithm('ExtractMask', InputWorkspace=ws_name, OutputWorkspace=mask_name)
masked_ws = AnalysisDataService[mask_name]
self.assertTrue(isinstance(masked_ws, IMaskWorkspace))
self.assertEqual(1, masked_ws.getNumberMasked())
# single number
self.assertTrue(not masked_ws.isMasked(1))
self.assertTrue(masked_ws.isMasked(2))
# list
self.assertTrue(not masked_ws.isMasked([1]))
self.assertTrue(masked_ws.isMasked([2]))
def test_MaskWorkspace_Is_Retrievable(self):
dummy_ws = WorkspaceCreationHelper.create2DWorkspaceWithFullInstrument(2, 102, False) # no monitors
ws_name = "dummy"
AnalysisDataService.add(ws_name, dummy_ws)
run_algorithm('MaskDetectors', Workspace=ws_name, WorkspaceIndexList=1)
mask_name = 'mask_ws'
run_algorithm('ExtractMask', InputWorkspace=ws_name, OutputWorkspace=mask_name)
masked_ws = AnalysisDataService[mask_name]
self.assertTrue(isinstance(masked_ws, IMaskWorkspace))
self.assertEqual(1, masked_ws.getNumberMasked())
# single number
self.assertTrue(not masked_ws.isMasked(1))
self.assertTrue(masked_ws.isMasked(2))
# list
self.assertTrue(not masked_ws.isMasked([1]))
self.assertTrue(masked_ws.isMasked([2]))
def test_pickle_table_workspace(self):
from mantid.kernel import V3D
import pickle
table = TableWorkspace()
table.addColumn(type="int", name="index")
table.addColumn(type="str", name="value")
table.addColumn(type="V3D", name="position")
values = (1, '10', V3D(0, 0, 1))
table.addRow(values)
values = (2, '100', V3D(1, 0, 0))
table.addRow(values)
p = pickle.dumps(table)
table2 = pickle.loads(p)
self.assertEqual(table.toDict(), table2.toDict())
# Can we add it to the ADS
name = "test_pickle_table_workspace"
AnalysisDataService.add(name, table2)
self.assertTrue(name in AnalysisDataService)
AnalysisDataService.remove(name)
def test_pickle_table_workspace(self):
from mantid.kernel import V3D
import pickle
table = TableWorkspace()
table.addColumn(type="int",name="index")
table.addColumn(type="str",name="value")
table.addColumn(type="V3D",name="position")
values = (1, '10', V3D(0, 0, 1))
table.addRow(values)
values = (2, '100', V3D(1, 0, 0))
table.addRow(values)
p = pickle.dumps(table)
table2 = pickle.loads(p)
self.assertEqual(table.toDict(), table2.toDict())
# Can we add it to the ADS
name = "test_pickle_table_workspace"
AnalysisDataService.add(name, table2)
self.assertTrue(name in AnalysisDataService)
AnalysisDataService.remove(name)
def _createMultiSpectra(self, wkspname):
wksp = WCH.create2DWorkspaceWithFullInstrument(30, 5, False, False)
AnalysisDataService.add(wkspname, wksp)
def _createMultiSpectra(self, wkspname):
wksp = WCH.create2DWorkspaceWithFullInstrument(30, 5, False, False)
AnalysisDataService.add(wkspname, wksp)
