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 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_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))))
