Skip to content

Commit

Permalink
Refs #10250 Re-add Stitch1DMany python test
Browse files Browse the repository at this point in the history
  • Loading branch information
Harry Jeffery committed Sep 17, 2014
1 parent 5e184bf commit a2c9bcc
Show file tree
Hide file tree
Showing 2 changed files with 161 additions and 0 deletions.
1 change: 1 addition & 0 deletions Code/Mantid/Framework/PythonInterface/test/python/plugins/algorithms/CMakeLists.txt
View file
Expand Up @@ -34,6 +34,7 @@ set ( TEST_PY_FILES
SofQWMomentsTest.py
SortDetectorsTest.py
SortXAxisTest.py
Stitch1DManyTest.py
SuggestTibCNCSTest.py
SuggestTibHYSPECTest.py
UpdatePeakParameterTableValueTest.py
Expand Down
160 changes: 160 additions & 0 deletions Code/Mantid/Framework/PythonInterface/test/python/plugins/algorithms/Stitch1DManyTest.py
View file
@@ -0,0 +1,160 @@
import unittest
import numpy
from mantid.simpleapi import *
from mantid.kernel import *
from mantid.api import *

class Stitch1DManyTest(unittest.TestCase):

a = None
b = None
c = None
x = None
e = None

def setUp(self):
x = numpy.arange(-1, 1.2, 0.2)
e = numpy.arange(-1, 1, 0.2)
e.fill(0)
self.e = e
self.x = x
a = CreateWorkspace(UnitX="1/q", DataX=x, DataY=[0.0,0.0,0.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0], NSpec=1, DataE=e)
b = CreateWorkspace(UnitX="1/q", DataX=x, DataY=[2.0,2.0,2.0,2.0,2.0,2.0,2.0,0.0,0.0,0.0], NSpec=1, DataE=e)
self.a = a
self.b = b

def tearDown(self):
# Cleanup
DeleteWorkspace(self.a)
DeleteWorkspace(self.b)

def do_check_ydata(self, expectedYData, targetWS):
print "expected"
print expectedYData
print "target"
print targetWS.readY(0)
yDataRounded = [ round(elem, 4) for elem in targetWS.readY(0) ]
same = all([(x == y) for x,y in zip(yDataRounded, expectedYData)])
self.assertTrue(same)


def test_stitch_throws_with_too_few_workspaces(self):
try:
stitched = Stitch1DMany(InputWorkspaces='a', StartOverlaps=[-0.5], EndOverlaps=[0.5], Params=[0.1])
self.fail("Only one workspace. Should have thrown.")
except RuntimeError:
pass

def test_stitch_throws_with_wrong_number_of_Start_overlaps(self):
try:
stitched = Stitch1DMany(InputWorkspaces='a, b', StartOverlaps=[-0.5, -0.6], EndOverlaps=[0.5], Params=[0.1])
self.fail("Two start overlaps, but only two workspaces. Should have thrown.")
except RuntimeError:
pass

def test_stitch_throws_with_wrong_number_of_End_overlaps(self):
try:
stitched = Stitch1DMany(InputWorkspaces='a, b', StartOverlaps=[-0.5], EndOverlaps=[0.5, 0.6], Params=[0.1])
self.fail("Two end overlaps, but only two workspaces. Should have thrown.")
except RuntimeError:
pass

def test_stich_throws_if_no_params(self):
try:
stitched = Stitch1DMany(InputWorkspaces='a, b')
self.fail("No Params given. Should have thrown.")
except RuntimeError:
pass

def test_workspace_types_differ_throws(self):
tbl = CreateEmptyTableWorkspace()
input_workspaces = "%s, %s" % (self.a.name(), tbl.name()) # One table workspace, one matrix workspace
try:
stitchedViaStitchMany, scaleFactorMany = Stitch1DMany(InputWorkspaces=input_workspaces, Params=0.2)
self.fail("Input workspace type mis-match. Should have thrown.")
except RuntimeError:
pass
finally:
DeleteWorkspace(tbl)

def test_workspace_group_size_differ_throws(self):
ws1 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[3.0, 3.0, 3.0, 3.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws2 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws3 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 1.0], NSpec=1, DataE=self.e)
input_group_1 = GroupWorkspaces(InputWorkspaces="%s,%s, %s" % (ws1.name(), ws2.name(), ws3.name()))
input_group_2 = GroupWorkspaces(InputWorkspaces="%s,%s" % (ws1.name(), ws2.name()))
try:
stitched, sf = Stitch1DMany(InputWorkspaces='%s,%s' % (input_group_1.name(), input_group_2.name()), Params=0.2)
self.fail("Differing number of sub-workspaces in workspace group. Should have thrown.")
except RuntimeError:
pass
finally:
DeleteWorkspace(input_group_1)

#Cross-check that the result of using Stitch1DMany with two workspaces is the same as using Stitch1D.

def test_stitches_two(self):
stitchedViaStitchMany, scaleFactorMany = Stitch1DMany(InputWorkspaces='a, b', StartOverlaps=[-0.4], EndOverlaps=[0.4], Params=[0.2])
stitchedViaStitchTwo, scaleFactorTwo = Stitch1D(LHSWorkspace=self.a, RHSWorkspace=self.b, StartOverlap=-0.4, EndOverlap=0.4, Params=[0.2])
self.assertTrue(isinstance(scaleFactorMany, numpy.ndarray), "Should be returning a list of scale factors")
self.assertEqual(1, scaleFactorMany.size)
self.assertEquals(scaleFactorTwo, scaleFactorMany[0])

expectedYData = [0,0,0,3,3,3,3,0,0,0]
self.do_check_ydata(expectedYData, stitchedViaStitchMany)

# Do cross compare
isSuccess = CheckWorkspacesMatch(Workspace1=stitchedViaStitchMany, Workspace2=stitchedViaStitchTwo)
self.assertEquals("Success!", isSuccess);

DeleteWorkspace(stitchedViaStitchMany)
DeleteWorkspace(stitchedViaStitchTwo)


def test_stitches_three(self):
ws1 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[3.0, 3.0, 3.0, 3.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws2 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws3 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 1.0], NSpec=1, DataE=self.e)
stitchedViaStitchMany, sfs = Stitch1DMany(InputWorkspaces='ws1, ws2, ws3', StartOverlaps=[-0.4,0.2], EndOverlaps=[-0.2,0.4], Params=0.2)

expectedYData = [3,3,3,3,3,3,3,3,3,3]
self.do_check_ydata(expectedYData, stitchedViaStitchMany)
self.assertEqual(2, sfs.size)
self.assertEquals(1.5, round(sfs[0], 6))
self.assertEquals(3.0, round(sfs[1], 6))

DeleteWorkspace(ws1)
DeleteWorkspace(ws2)
DeleteWorkspace(ws3)
DeleteWorkspace(stitchedViaStitchMany)

def test_stitches_three_no_overlaps_specified_should_still_work(self):
ws1 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[3.0, 3.0, 3.0, 3.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws2 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws3 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 1.0], NSpec=1, DataE=self.e)
stitchedViaStitchMany, sf = Stitch1DMany(InputWorkspaces='ws1, ws2, ws3', Params=0.2)


def test_stitches_using_manual_scaling(self):
stitchedViaStitchMany, sf = Stitch1DMany(InputWorkspaces='a, b', StartOverlaps=[-0.4], EndOverlaps=[0.4], Params=[0.2], UseManualScaleFactor=True, ManualScaleFactor=2.0)

self.assertEquals(2.0, round(sf, 6))
DeleteWorkspace(stitchedViaStitchMany)

def test_process_group_workspaces(self):
ws1 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[3.0, 3.0, 3.0, 3.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws2 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 0.0, 0.0, 0.0], NSpec=1, DataE=self.e)
ws3 = CreateWorkspace(UnitX="1/q", DataX=self.x, DataY=[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 1.0], NSpec=1, DataE=self.e)
input_group_1 = GroupWorkspaces(InputWorkspaces="%s,%s,%s" % (ws1.name(), ws2.name(), ws3.name()))
input_group_2 = GroupWorkspaces(InputWorkspaces="%s,%s,%s" % (ws1.name(), ws2.name(), ws3.name()))
stitched, sfs = Stitch1DMany(InputWorkspaces='%s,%s' % (input_group_1.name(), input_group_2.name()), Params=0.2)
self.assertTrue(isinstance(stitched, WorkspaceGroup), "Output should be a group workspace")
self.assertEqual(stitched.size(), 3, "Output should contain 3 workspaces")
self.assertEqual(stitched.name(), "stitched", "Output not named correctly")
self.assertEquals(input_group_1.size(), sfs.size)
DeleteWorkspace(input_group_1)



if __name__ == '__main__':
unittest.main()

0 comments on commit a2c9bcc

Please sign in to comment.