def test_GIVEN_angled_mirror_is_disabled_WHEN_get_beam_out_THEN_outgoing_beam_is_incoming_beam(
self):
mirror_z_position = 10
mirror_angle = 15
beam_start = PositionAndAngle(y=0, z=0, angle=0)
expected = beam_start
mirror = ReflectingComponent("component",
movement_strategy=LinearMovement(
0, mirror_z_position, 90))
mirror.angle = mirror_angle
mirror.set_incoming_beam(beam_start)
mirror.enabled = False
result = mirror.get_outgoing_beam()
assert_that(result, is_(position_and_angle(expected)))
def test_GIVEN_mirror_with_input_beam_at_WHEN_get_beam_out_THEN_beam_output_correct(
self, beam_angle, mirror_angle, outgoing_angle):
beam_start = PositionAndAngle(y=0, z=0, angle=beam_angle)
expected = PositionAndAngle(y=0, z=0, angle=outgoing_angle)
mirror = ReflectingComponent("component",
movement_strategy=LinearMovement(
0, 0, 90))
mirror.angle = mirror_angle
mirror.set_incoming_beam(beam_start)
result = mirror.get_outgoing_beam()
assert_that(
result, is_(position_and_angle(expected)),
"beam_angle: {}, mirror_angle: {}".format(beam_angle,
mirror_angle))
def test_GIVEN_mirror_with_input_beam_at_0_deg_and_z0_y0_WHEN_get_beam_out_THEN_beam_output_z_is_zmirror_y_is_ymirror_angle_is_input_angle_plus_device_angle(
self):
mirror_z_position = 10
mirror_angle = 15
beam_start = PositionAndAngle(y=0, z=0, angle=0)
expected = PositionAndAngle(y=0,
z=mirror_z_position,
angle=2 * mirror_angle)
mirror = ReflectingComponent("component",
movement_strategy=LinearMovement(
0, mirror_z_position, 90))
mirror.angle = mirror_angle
mirror.set_incoming_beam(beam_start)
result = mirror.get_outgoing_beam()
assert_that(result, is_(position_and_angle(expected)))
class TestComponentBeamline(unittest.TestCase):
def setUp(self):
beam_start = PositionAndAngle(y=0, z=0, angle=2.5)
s0 = Component("s0", movement_strategy=LinearMovement(0, 0, 90))
s1 = Component("s1", movement_strategy=LinearMovement(0, 1, 90))
frame_overlap_mirror = ReflectingComponent(
"FOM", movement_strategy=LinearMovement(0, 2, 90))
frame_overlap_mirror.enabled = False
self.polarising_mirror = ReflectingComponent(
"Polariser", movement_strategy=LinearMovement(0, 3, 90))
self.polarising_mirror.enabled = False
self.polarising_mirror.angle = 0
s2 = Component("s2", movement_strategy=LinearMovement(0, 4, 90))
self.ideal_sample_point = ReflectingComponent(
"ideal sample point", movement_strategy=LinearMovement(0, 5, 90))
s3 = Component("s3", movement_strategy=LinearMovement(0, 6, 90))
analyser = ReflectingComponent("analyser",
movement_strategy=LinearMovement(
0, 7, 90))
analyser.enabled = False
s4 = Component("s4", movement_strategy=LinearMovement(0, 8, 90))
detector = Component("detector",
movement_strategy=LinearMovement(0, 10, 90))
theta = Theta("theta", self.ideal_sample_point)
theta.sp_no_move = 0
smangle = ReflectionAngle("smangle", self.polarising_mirror)
smangle.sp_no_move = 0
self.nr_mode = BeamlineMode("NR Mode", [theta.name])
self.polarised_mode = BeamlineMode("NR Mode",
[smangle.name, theta.name])
self.beamline = Beamline([
s0, s1, frame_overlap_mirror, self.polarising_mirror, s2,
self.ideal_sample_point, s3, analyser, s4, detector
], [smangle, theta], [], [self.nr_mode, self.polarised_mode])
self.beamline.set_incoming_beam(beam_start)
def test_GIVEN_beam_line_contains_multiple_component_WHEN_set_theta_THEN_angle_between_incoming_and_outgoing_beam_is_correct(
self):
self.beamline.active_mode = self.nr_mode
theta_set = 10.0
self.beamline.parameter("theta").sp = theta_set
reflection_angle = self.ideal_sample_point.get_outgoing_beam(
).angle - self.ideal_sample_point.incoming_beam.angle
assert_that(reflection_angle, is_(theta_set * 2.0))
def test_GIVEN_beam_line_contains_active_super_mirror_WHEN_set_theta_THEN_angle_between_incoming_and_outgoing_beam_is_correct(
self):
self.beamline.active_mode = self.polarised_mode
theta_set = 10.0
self.polarising_mirror.enabled = True
self.beamline.parameter("smangle").sp = 10
self.beamline.parameter("theta").sp = theta_set
reflection_angle = self.ideal_sample_point.get_outgoing_beam(
).angle - self.ideal_sample_point.incoming_beam.angle
assert_that(reflection_angle, is_(theta_set * 2.0))
def test_GIVEN_beam_line_contains_active_super_mirror_WHEN_angle_set_THEN_angle_between_incoming_and_outgoing_beam_is_correct(
self):
self.beamline.active_mode = self.polarised_mode
theta_set = 10.0
self.beamline.parameter("theta").sp = theta_set
self.polarising_mirror.enabled = True
self.beamline.parameter("smangle").sp = 10
reflection_angle = self.ideal_sample_point.get_outgoing_beam(
).angle - self.ideal_sample_point.incoming_beam.angle
assert_that(reflection_angle, is_(theta_set * 2.0))