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)