Commit 069604c7 authored by Owen Arnold's avatar Owen Arnold
Browse files

refs #7472. Adding tests for getGroup.

Lots of options for using getGroup. Cover the specific 'whattoget' arguments.
parent b4b3b510
import unittest import unittest
import numpy
from MantidFramework import mtd from MantidFramework import mtd
mtd.initialise() mtd.initialise()
from mantid.simpleapi import * from mantid.simpleapi import *
...@@ -17,12 +18,15 @@ class ReflectometryQuickAuxiliaryTest(unittest.TestCase): ...@@ -17,12 +18,15 @@ class ReflectometryQuickAuxiliaryTest(unittest.TestCase):
self.assertEqual(1, len(cleaned_object_names)) self.assertEqual(1, len(cleaned_object_names))
self.assertEqual(True, ('tokeep' in cleaned_object_names)) self.assertEqual(True, ('tokeep' in cleaned_object_names))
DeleteWorkspace(tokeep)
def test_coAdd_ws_in_ADS(self): def test_coAdd_ws_in_ADS(self):
inWS = CreateSingleValuedWorkspace(DataValue=1, ErrorValue=1) inWS = CreateSingleValuedWorkspace(DataValue=1, ErrorValue=1)
quick.coAdd('inWS', 'ProvidedName') quick.coAdd('inWS', 'ProvidedName')
outWS = mtd['_WProvidedName'] outWS = mtd['_WProvidedName']
result = CheckWorkspacesMatch(Workspace1=inWS, Workspace2=outWS) result = CheckWorkspacesMatch(Workspace1=inWS, Workspace2=outWS)
self.assertEquals("Success!", result) self.assertEquals("Success!", result)
DeleteWorkspace(outWS)
def test_coAdd_run_list(self): def test_coAdd_run_list(self):
originalInstrument = config.getInstrument() originalInstrument = config.getInstrument()
...@@ -49,7 +53,74 @@ class ReflectometryQuickAuxiliaryTest(unittest.TestCase): ...@@ -49,7 +53,74 @@ class ReflectometryQuickAuxiliaryTest(unittest.TestCase):
self.assertEquals("Success!", result) self.assertEquals("Success!", result)
finally: finally:
config['default.instrument'] = originalInstrument.name() config['default.instrument'] = originalInstrument.name()
DeleteWorkspace(a[0])
DeleteWorkspace(b[0])
DeleteWorkspace(c)
DeleteWorkspace(outWS)
def test_groupGet_instrument(self):
wsName = "TestWorkspace"
LoadISISNexus(Filename='POLREF00004699', OutputWorkspace=wsName)
expectedInstrument = "POLREF"
# Test with group workspace as input
instrument = quick.groupGet(wsName, 'inst')
self.assertEquals(expectedInstrument, instrument.getName(), "Did not fetch the instrument from ws group")
# Test with single workspace as input
instrument = quick.groupGet(mtd[wsName][0].name(), 'inst')
self.assertEquals(expectedInstrument, instrument.getName(), "Did not fetch the instrument from ws")
DeleteWorkspace(mtd[wsName])
def test_groupGet_histogram_count(self):
wsName = "TestWorkspace"
LoadISISNexus(Filename='POLREF00004699', OutputWorkspace=wsName)
expectedNHistograms = mtd[wsName][0].getNumberHistograms()
# Test with group workspace as input
nHistograms = quick.groupGet(wsName, 'wksp')
self.assertEquals(expectedNHistograms, nHistograms, "Did not fetch the n histograms from ws group")
# Test with single workspace as input
nHistograms = quick.groupGet(mtd[wsName][0].name(), 'wksp')
self.assertEquals(expectedNHistograms, nHistograms, "Did not fetch the n histograms from ws")
DeleteWorkspace(mtd[wsName])
def test_groupGet_log_single_value(self):
wsName = "TestWorkspace"
LoadISISNexus(Filename='POLREF00004699', OutputWorkspace=wsName)
expectedNPeriods = 2
# Test with group workspace as input
nPeriods = quick.groupGet(wsName, 'samp', 'nperiods')
self.assertEquals(expectedNPeriods, nPeriods, "Did not fetch the number of periods from ws group")
# Test with single workspace as input
nPeriods = quick.groupGet(mtd[wsName][0].name(), 'samp', 'nperiods')
self.assertEquals(expectedNPeriods, nPeriods, "Did not fetch the number of periods from ws")
DeleteWorkspace(mtd[wsName])
def test_groupGet_multi_value_log(self):
wsName = "TestWorkspace"
LoadISISNexus(Filename='POLREF00004699', OutputWorkspace=wsName)
# Expected start theta, taken from the last value of the time series log.
expectedStartTheta = 0.4903
# Test with group workspace as input
stheta = quick.groupGet(wsName, 'samp', 'stheta')
self.assertEquals(expectedStartTheta, round(float(stheta), 4))
# Test with single workspace as input
stheta = quick.groupGet(mtd[wsName][0].name(), 'samp', 'stheta')
self.assertEquals(expectedStartTheta, round(float(stheta), 4))
DeleteWorkspace(mtd[wsName])
if __name__ == '__main__': if __name__ == '__main__':
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment