def testReflectedBeamSumInQExecutes(self): dirWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.add_chopper_configuration_D17(dirWS) dirBeamPosWS = illhelpers.refl_create_beam_position_ws('dirBeamPosWS', dirWS, 0., 128) dirWS = illhelpers.refl_preprocess('dirWS', dirWS, dirBeamPosWS) args = { 'InputWorkspace': dirWS, 'OutputWorkspace': 'dirForeground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted()) dirForeground = alg.getProperty('OutputWorkspace').value reflWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(reflWS, 1.2) illhelpers.add_chopper_configuration_D17(reflWS) reflBeamPosWS = illhelpers.refl_create_beam_position_ws('reflBeamPosWS', reflWS, 1.2, 128) reflWS = illhelpers.refl_preprocess('refWS', reflWS, reflBeamPosWS) args = { 'InputWorkspace': reflWS, 'OutputWorkspace': 'foreground', 'DirectForegroundWorkspace': dirForeground, 'SummationType': 'SumInQ', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted())
def testSumInQModeProducesDX(self): dirWS = illhelpers.create_poor_mans_d17_workspace() mtd.add('dirWS', dirWS) illhelpers.add_slit_configuration_D17(dirWS, 0.03, 0.02) illhelpers.add_chopper_configuration_D17(dirWS) illhelpers.refl_create_beam_position_ws('dirBeamPosWS', dirWS, 0., 128) dirWS = illhelpers.refl_preprocess('dirWS', dirWS, 'dirBeamPosWS') dirFgdWS = illhelpers.refl_sum_foreground('dirFgdWS', 'SumInLambda', dirWS) reflWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.add_chopper_configuration_D17(reflWS) illhelpers.add_slit_configuration_D17(reflWS, 0.03, 0.02) illhelpers.refl_rotate_detector(reflWS, 1.5) mtd.add('reflWS', reflWS) illhelpers.refl_create_beam_position_ws('reflBeamPosWS', reflWS, 1.5, 128) reflWS = illhelpers.refl_preprocess('reflWS', reflWS, 'reflBeamPosWS') fgdWS = illhelpers.refl_sum_foreground('fgdWS', 'SumInQ', reflWS, dirFgdWS, dirWS) args = { 'InputWorkspace': fgdWS, 'OutputWorkspace': 'inQ', 'DirectForegroundWorkspace': dirFgdWS, 'GroupingQFraction': 0.2, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLConvertToQ', **args) assertRaisesNothing(self, alg.execute) outWS = alg.getProperty('OutputWorkspace').value self.assertEqual(outWS.getNumberHistograms(), 1) self.assertTrue(outWS.hasDx(0))
def testReflectedBeamSumInQExecutes(self): dirWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.add_chopper_configuration_D17(dirWS) illhelpers.add_slit_configuration_D17(dirWS, 0.02, 0.03) dirBeamPosWS = illhelpers.refl_create_beam_position_ws('dirBeamPosWS', dirWS, 0., 128) dirWS = illhelpers.refl_preprocess('dirWS', dirWS, dirBeamPosWS) args = { 'InputWorkspace': dirWS, 'OutputWorkspace': 'dirForeground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted()) dirForeground = alg.getProperty('OutputWorkspace').value reflWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(reflWS, 1.2) illhelpers.add_chopper_configuration_D17(reflWS) illhelpers.add_slit_configuration_D17(reflWS, 0.02, 0.03) reflBeamPosWS = illhelpers.refl_create_beam_position_ws('reflBeamPosWS', reflWS, 1.2, 128) reflWS = illhelpers.refl_preprocess('refWS', reflWS, reflBeamPosWS) args = { 'InputWorkspace': reflWS, 'OutputWorkspace': 'foreground', 'DirectForegroundWorkspace': dirForeground, 'SummationType': 'SumInQ', 'DirectBeamWorkspace': dirWS, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted())
def testExecutes(self): dirWS = illhelpers.create_poor_mans_d17_workspace() mtd.add('dirWS', dirWS) illhelpers.add_slit_configuration_D17(dirWS, 0.03, 0.02) illhelpers.refl_create_beam_position_ws('dirBeamPosWS', dirWS, 0., 128) dirWS = illhelpers.refl_preprocess('dirWS', dirWS, 'dirBeamPosWS') reflWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.add_chopper_configuration_D17(reflWS) illhelpers.add_slit_configuration_D17(reflWS, 0.03, 0.02) illhelpers.refl_rotate_detector(reflWS, 1.5) mtd.add('reflWS', reflWS) illhelpers.refl_create_beam_position_ws('reflBeamPosWS', reflWS, 1.5, 128) reflWS = illhelpers.refl_preprocess('reflWS', reflWS, 'reflBeamPosWS') fgdWS = illhelpers.refl_sum_in_lambda('fgdWS', reflWS) args = { 'InputWorkspace': fgdWS, 'OutputWorkspace': 'inQ', 'ReflectedBeamWorkspace': reflWS, 'DirectBeamWorkspace': dirWS, 'GroupingQFraction': 0.2, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLConvertToQ', **args) assertRaisesNothing(self, alg.execute)
def testDirectBeamSummationExecutes(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) beamPosWS = illhelpers.refl_create_beam_position_ws('beamPosWS', ws, 1.2, 128) ws = illhelpers.refl_preprocess('ws', ws, beamPosWS) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted())
def testReflectedBeamSumInLambdaDetectorMovingAndRotation(self): dirWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.add_chopper_configuration_D17(dirWS) illhelpers.add_slit_configuration_D17(dirWS, 0.03, 0.02) dirWS = illhelpers.refl_add_two_theta(dirWS, 6.7) dirWS = illhelpers.refl_preprocess_lineposition('dirWS', dirWS, 50.9) self.assertEqual(dirWS.run().getProperty(common.SampleLogs.TWO_THETA).value, 6.7) self.assertEqual(dirWS.run().getProperty(common.SampleLogs.LINE_POSITION).value, 50.9) args = { 'InputWorkspace': dirWS, 'OutputWorkspace': 'dirForeground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted()) dirForeground = alg.getProperty('OutputWorkspace').value self.assertEqual(dirForeground.run().getProperty(common.SampleLogs.TWO_THETA).value, 6.7) self.assertEqual(dirForeground.run().getProperty(common.SampleLogs.LINE_POSITION).value, 50.9) self.assertEqual(dirForeground.spectrumInfo().size(), 1) self.assertAlmostEqual(dirForeground.spectrumInfo().l2(0), 3.101234371122588, delta=1.e-15) self.assertAlmostEqual(numpy.degrees(dirForeground.spectrumInfo().twoTheta(0)), 8.3914043569830, delta=1.e-13) reflWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(reflWS, 1.2) illhelpers.add_chopper_configuration_D17(reflWS) illhelpers.add_slit_configuration_D17(reflWS, 0.03, 0.02) reflWS = illhelpers.refl_add_two_theta(reflWS, 40.2) reflWS = illhelpers.refl_preprocess_lineposition('refWS', reflWS, 120.4) self.assertEqual(reflWS.run().getProperty(common.SampleLogs.TWO_THETA).value, 40.2) self.assertEqual(reflWS.run().getProperty(common.SampleLogs.LINE_POSITION).value, 120.4) args = { 'InputWorkspace': reflWS, 'OutputWorkspace': 'foreground', 'DirectForegroundWorkspace': dirForeground, 'SummationType': 'SumInLambda', 'DirectLineWorkspace': dirWS, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) out = alg.getProperty('OutputWorkspace').value self.assertEqual(out.run().getProperty(common.SampleLogs.TWO_THETA).value, 40.2) self.assertEqual(out.run().getProperty(common.SampleLogs.LINE_POSITION).value, 120.4) self.assertEqual(out.spectrumInfo().size(), 1) self.assertAlmostEqual(out.spectrumInfo().l2(0), 3.0992766423566, delta=1.e-13) self.assertAlmostEqual(numpy.degrees(out.spectrumInfo().twoTheta(0)), 40.3568485178340, delta=1.e-13) self.assertTrue(alg.isExecuted())
def testNotSummedDirectForegroundRaises(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) beamPosWS = illhelpers.refl_create_beam_position_ws('beamPosWS', ws, 1.2, 128) ws = illhelpers.refl_preprocess('ws', ws, beamPosWS) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'DirectForegroundWorkspace': ws, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) self.assertRaisesRegexp(RuntimeError, 'Some invalid Properties found', alg.execute) self.assertTrue(alg.isExecuted)
def testNotSummedDirectForegroundRaises(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) ws = illhelpers.refl_add_two_theta(ws, 5.5) ws = illhelpers.refl_preprocess('ws', ws) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'DirectForegroundWorkspace': ws, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) self.assertRaisesRegex(RuntimeError, 'Some invalid Properties found', alg.execute) self.assertTrue(alg.isExecuted)
def testNoDirectForegroundAndSumInQRaises(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) beamPosWS = illhelpers.refl_create_beam_position_ws( 'beamPosWS', ws, 1.2, 128) ws = illhelpers.refl_preprocess('ws', ws, beamPosWS) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'SummationType': 'SumInQ', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) self.assertRaisesRegexp(RuntimeError, 'Some invalid Properties found', alg.execute) self.assertTrue(alg.isExecuted)
def testWavelengthRangeDefault(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) ws = illhelpers.refl_add_two_theta(ws, 5.5) ws = illhelpers.refl_preprocess('ws', ws) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) out = alg.getProperty('OutputWorkspace').value self.assertEqual(out.getNumberHistograms(), 1) Xs = out.readX(0) self.assertGreater(len(Xs), 1) self.assertGreater(Xs[0], 0.) self.assertLess(Xs[-1], 30.)
def testWavelengthRangeDefault(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) beamPosWS = illhelpers.refl_create_beam_position_ws('beamPosWS', ws, 1.2, 128) ws = illhelpers.refl_preprocess('ws', ws, beamPosWS) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) out = alg.getProperty('OutputWorkspace').value self.assertEquals(out.getNumberHistograms(), 1) Xs = out.readX(0) self.assertGreater(len(Xs), 1) self.assertGreater(Xs[0], 0.) self.assertLess(Xs[-1], 30.)
def testLinePositionFromSampleLogs(self): # We do not run the preprocess! ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) ws.getAxis(0).setUnit('Wavelength') ws = illhelpers.refl_add_two_theta(ws, 5.5) ws = illhelpers.refl_add_line_position(ws, 120.6) ws.run().addProperty('reduction.foreground.first_workspace_index', int(140), True) ws.run().addProperty('reduction.foreground.centre_workspace_index', int(150), True) ws.run().addProperty('reduction.foreground.last_workspace_index', int(160), True) args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted())
def testWavelengthRange(self): ws = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(ws, 1.2) beamPosWS = illhelpers.refl_create_beam_position_ws('beamPosWS', ws, 1.2, 128) ws = illhelpers.refl_preprocess('ws', ws, beamPosWS) xMin = 2.3 xMax = 4.2 args = { 'InputWorkspace': ws, 'OutputWorkspace': 'foreground', 'WavelengthRange': [xMin, xMax], 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) ws = alg.getProperty('OutputWorkspace').value self.assertEquals(ws.getNumberHistograms(), 1) Xs = ws.readX(0) self.assertGreater(len(Xs), 1) self.assertGreater(Xs[0], xMin) self.assertLess(Xs[-1], xMax)
def testReflectedBeamSumInLambdaExecutes(self): dirWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.add_chopper_configuration_D17(dirWS) illhelpers.add_slit_configuration_D17(dirWS, 0.03, 0.02) dirWS = illhelpers.refl_add_line_position(dirWS, 128.0) dirWS = illhelpers.refl_add_two_theta(dirWS, 6.7) dirWS = illhelpers.refl_preprocess('dirWS', dirWS) args = { 'InputWorkspace': dirWS, 'OutputWorkspace': 'dirForeground', 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted()) dirForeground = alg.getProperty('OutputWorkspace').value reflWS = illhelpers.create_poor_mans_d17_workspace() illhelpers.refl_rotate_detector(reflWS, 1.2) illhelpers.add_chopper_configuration_D17(reflWS) illhelpers.add_slit_configuration_D17(reflWS, 0.03, 0.02) reflWS = illhelpers.refl_add_line_position(reflWS, 128.0) reflWS = illhelpers.refl_add_two_theta(reflWS, 6.7) reflWS = illhelpers.refl_preprocess('refWS', reflWS) args = { 'InputWorkspace': reflWS, 'OutputWorkspace': 'foreground', 'DirectForegroundWorkspace': dirForeground, 'SummationType': 'SumInLambda', 'DirectLineWorkspace': dirWS, 'rethrow': True, 'child': True } alg = create_algorithm('ReflectometryILLSumForeground', **args) assertRaisesNothing(self, alg.execute) self.assertTrue(alg.isExecuted())