def test_removeFromGroup_removes_workspace_from_group(self):
from mantid.simpleapi import CreateSampleWorkspace, GroupWorkspaces
CreateSampleWorkspace(OutputWorkspace="ws1")
CreateSampleWorkspace(OutputWorkspace="ws2")
CreateSampleWorkspace(OutputWorkspace="ws3")
GroupWorkspaces(InputWorkspaces="ws1,ws2,ws3", OutputWorkspace="NewGroup")
AnalysisDataService.removeFromGroup("NewGroup", "ws3")
group = mtd['NewGroup']
self.assertEqual(group.size(), 2)
self.assertCountEqual(group.getNames(), ["ws1", "ws2"])
def test_removeFromGroup_removes_workspace_from_group(self):
from mantid.simpleapi import CreateSampleWorkspace, GroupWorkspaces
CreateSampleWorkspace(OutputWorkspace="ws1")
CreateSampleWorkspace(OutputWorkspace="ws2")
CreateSampleWorkspace(OutputWorkspace="ws3")
GroupWorkspaces(InputWorkspaces="ws1,ws2,ws3", OutputWorkspace="NewGroup")
AnalysisDataService.removeFromGroup("NewGroup", "ws3")
group = mtd['NewGroup']
self.assertEquals(group.size(), 2)
six.assertCountEqual(self, group.getNames(), ["ws1", "ws2"])
