def _init_beam_path_calcs(self, setup):
self._beam_path_set_point = SettableBeamPathCalcWithAngle(
"{}_sp".format(self.name),
LinearMovementCalc(setup),
is_reflecting=True)
self._beam_path_rbv = SettableBeamPathCalcWithAngle(
"{}_rbv".format(self.name),
LinearMovementCalc(setup),
is_reflecting=True)
def test_GIVEN_movement_45_to_z_at_beam_angle_along_z_WHEN_get_intercept_THEN_position_is_initial_position(
self, angle):
y = 0
z = 10
movement = LinearMovementCalc(PositionAndAngle(y, z, 45))
beam = PositionAndAngle(0, 0, angle)
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(y, z))))
def test_GIVEN_movement_perpendicular_to_z_at_beam_angle_0_WHEN_get_intercept_THEN_position_is_initial_position(
self):
y = 0
z = 10
movement = LinearMovementCalc(PositionAndAngle(y, z, 90))
beam = PositionAndAngle(0, 0, 0)
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(y, z))))
def test_GIVEN_set_point_value_set_WHEN_there_is_a_value_observer_THEN_observer_triggered_with_new_value(
self, expected_value):
movement = LinearMovementCalc(PositionAndAngle(0, 0, 90))
beam = PositionAndAngle(0, 0, 0)
self._value = None
movement.set_displacement(expected_value)
result = movement.get_distance_relative_to_beam(beam)
assert_that(result, is_(expected_value))
def test_GIVEN_displacement_WHEN_calculating_position_in_mantid_coordinates_THEN_coordinates_at_given_displacement_are_returned(
self):
y = 0
z = 10
angle = 90
movement = LinearMovementCalc(PositionAndAngle(y, z, angle))
displacement = 5
result = movement.position_in_mantid_coordinates(displacement)
assert_that(result, is_(position(Position(displacement, z))))
def test_GIVEN_movement_anti_perpendicular_to_z_at_beam_angle_10_WHEN_get_intercept_THEN_position_is_z_as_initial_y_as_right_angle_triangle(
self):
y = 0
z = 10
beam_angle = 10
movement = LinearMovementCalc(PositionAndAngle(y, z, -90))
beam = PositionAndAngle(0, 0, beam_angle)
expected_y = z * tan(radians(beam_angle))
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(expected_y, z))))
def test_GIVEN_movement_0_to_z_and_beam_angle_45_WHEN_get_intercept_THEN_position_is_on_movement_axis(
self):
y = 20
z = 10
movement = LinearMovementCalc(PositionAndAngle(y, z, 0))
beam = PositionAndAngle(0, 0, 45)
expected_y = y
expected_z = y
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(expected_y, expected_z))))
def test_GIVEN_movement_along_z_WHEN_set_position_relative_to_beam_to_10_THEN_position_is_at_10_along_intercept(
self):
movement = LinearMovementCalc(PositionAndAngle(0, 10, 0))
beam_intercept = Position(0, 10)
beam = PositionAndAngle(0, 10, 270)
dist = 10
movement.set_distance_relative_to_beam(beam, dist)
result = movement.position_in_mantid_coordinates()
assert_that(
result,
is_(position(Position(beam_intercept.y, beam_intercept.z + dist))))
def test_GIVEN_movement_20_to_z_and_beam_angle_45_WHEN_get_intercept_THEN_position_is_on_movement_axis(
self):
beam = PositionAndAngle(0, 0, 45)
expected_y = 4
expected_z = 4
move_angle = 20
move_z = 2
move_y = expected_y - (expected_z - move_z) * tan(radians(move_angle))
movement = LinearMovementCalc(
PositionAndAngle(move_y, move_z, move_angle))
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(expected_y, expected_z))))
def test_GIVEN_movement_at_30_to_z_beam_intercept_is_at_zero_WHEN_set_position_relative_to_beam_to_10_THEN_position_is_at_10_along_intercept(
self, add_angle):
# here the beam intercept is above and to the right of the zero point
movement = LinearMovementCalc(PositionAndAngle(0, 10, 30 + add_angle))
beam = PositionAndAngle(0, 0, 0)
beam_intercept = Position(0, 10)
dist = 10
movement.set_distance_relative_to_beam(beam, dist)
result = movement.position_in_mantid_coordinates()
assert_that(
result,
is_(
position(
Position(beam_intercept.y + dist / 2.0,
beam_intercept.z + dist * sqrt(3) / 2.0))))
def _init_beam_path_calcs(self, setup):
linear_movement_calc = LinearMovementCalc(setup)
self._beam_path_set_point = BeamPathCalcThetaSP(
"{}_sp".format(self.name), linear_movement_calc)
self._beam_path_rbv = BeamPathCalcThetaRBV("{}_rbv".format(self.name),
linear_movement_calc,
self._beam_path_set_point)
def test_GIVEN_movement_and_beam_at_the_same_angle_WHEN_get_intercept_THEN_raises_calc_error(
self):
angle = 12.3
movement = LinearMovementCalc(PositionAndAngle(1, 1, angle))
beam = PositionAndAngle(0, 0, angle)
assert_that(
calling(movement.calculate_interception).with_args(beam),
raises(ValueError))
def test_GIVEN_movement_and_beam_at_the_opposite_angles_within_tolerance_WHEN_get_intercept_THEN_raises_calc_error(
self):
angle = 12.3 + 180.0
tolerance = ANGULAR_TOLERANCE
movement = LinearMovementCalc(
PositionAndAngle(1, 1, angle + tolerance * 0.99))
beam = PositionAndAngle(0, 0, angle)
assert_that(
calling(movement.calculate_interception).with_args(beam),
raises(ValueError))
def test_GIVEN_movement_45_to_z_at_beam_along_z_WHEN_z_of_movement_axis_changes_THEN_position_is_new_z(
self):
y = 0
z = 7
z_offset = 9
movement = LinearMovementCalc(PositionAndAngle(y, z, 45))
beam = PositionAndAngle(y, 0, 0)
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(y, z))))
movement.offset_position_at_zero(Position(0, z_offset))
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(y, z + z_offset))))
def test_GIVEN_movement_perp_to_z_at_beam_angle_45_WHEN_z_of_movement_axis_changes_THEN_position_is_as_expected(
self):
y = 0
z = 7
z_offset = 9
movement = LinearMovementCalc(PositionAndAngle(y, z, 90))
beam = PositionAndAngle(0, 0, 45)
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(z, z))))
movement.offset_position_at_zero(Position(0, z_offset))
result = movement.calculate_interception(beam)
assert_that(result, is_(position(Position(z + z_offset,
z + z_offset))))
def _init_beam_path_calcs(self, setup):
self._beam_path_set_point = TrackingBeamPathCalc(
"{}_sp".format(self.name), LinearMovementCalc(setup))
self._beam_path_rbv = TrackingBeamPathCalc("{}_rbv".format(self.name),
LinearMovementCalc(setup))
