def test_handler_validate_transform_mask():
"""Test."""
# test predesigned mask sequence
OptimizationHandler(robot=Robot.from_parameters(ur10()),
tool_mask=[False] * 6)
# test error
with raises(PyboticsError):
OptimizationHandler(robot=Robot.from_parameters(ur10()),
kinematic_chain_mask=[False])
def test_fk(resources_path: Path):
"""
Test robot.
:param robot:
:return:
"""
# get resource
path = resources_path / "ur10-joints-poses.csv"
data = np.loadtxt(str(path), delimiter=",")
# load robot
robot = Robot.from_parameters(ur10())
# test
for d in data:
n = robot.ndof
joints = np.deg2rad(d[:n])
desired_pose = d[n:].reshape((4, 4))
atol = 1e-3
# test with position argument
actual_pose = robot.fk(q=joints)
np.testing.assert_allclose(actual_pose, desired_pose, atol=atol)
# test with internal position attribute
robot.joints = joints
actual_pose = robot.fk()
np.testing.assert_allclose(actual_pose, desired_pose, atol=atol)
def test_optimization():
"""Test."""
# init robot model and error wrt nominal
actual_robot = Robot.from_parameters(ur10())
actual_robot.tool.position = [0.1, 0, 0]
actual_robot.kinematic_chain.links[0].a += 0.1
# calculate fk
qs = np.tile(np.linspace(start=-np.pi, stop=np.pi, num=100),
(len(ur10()), 1)).transpose()
poses = np.array(list(map(actual_robot.fk, qs)))
positions = poses[:, :-1, -1]
# init handler
handler = OptimizationHandler(robot=Robot.from_parameters(ur10()))
handler.kinematic_chain_mask[1] = True
handler.tool_mask[0] = True
# run optimization
result = scipy.optimize.least_squares(
fun=optimize_accuracy,
x0=handler.generate_optimization_vector(),
args=(handler, qs, positions),
verbose=2,
) # type: scipy.optimize.OptimizeResult
# validate
atol = 1e-2
np.testing.assert_allclose(actual=result.x,
desired=handler.generate_optimization_vector(),
atol=atol)
np.testing.assert_allclose(
actual=handler.robot.kinematic_chain.vector,
desired=actual_robot.kinematic_chain.vector,
atol=atol,
)
np.testing.assert_allclose(actual=handler.robot.tool.vector,
desired=actual_robot.tool.vector,
atol=atol)
np.testing.assert_allclose(actual=handler.robot.world_frame,
desired=actual_robot.world_frame,
atol=atol)
def test_joint_limits():
"""Test."""
robot = Robot.from_parameters(ur10())
# test setter
robot.joint_limits = robot.joint_limits.copy()
# test errors
with raises(PyboticsError):
robot.joint_limits = np.zeros(1)
with raises(PyboticsError):
robot.joints = robot.joint_limits.copy()[1] + 10
def test_compute_absolute_errors(q: np.ndarray):
"""Test."""
robot = Robot.from_parameters(ur10())
pose = robot.fk(q)
p = pose[:-1, -1]
# test 1D input
actual_error = compute_absolute_error(q=q, position=p, robot=robot)
np.testing.assert_allclose(actual_error, 0)
# test 2D input
actual_error = compute_absolute_errors(qs=np.tile(q, (10, 1)),
positions=np.tile(p, (10, 1)),
robot=robot)
np.testing.assert_allclose(actual_error, 0)
def main():
"""
Demonstrate pybotics usage.
View source for more info.
"""
# init robot
robot = Robot.from_parameters(ur10())
# add tool
tool = Tool()
tool.position = [1, 2, 3]
robot.tool = tool
# set world frame
world_frame = vector_2_matrix([100, 200, 300, 0, 0, 0])
robot.world_frame = world_frame
print(f"Robot: {robot}")
print(f"Kinematic Chain: {robot.kinematic_chain}")
def test_compute_relative_errors(q_a: np.ndarray, q_b: np.ndarray):
"""Test."""
robot = Robot.from_parameters(ur10())
p_a = robot.fk(q_a)[:-1, -1]
p_b = robot.fk(q_b)[:-1, -1]
distance = np.linalg.norm(p_a - p_b)
# test 1D input
actual_error = compute_relative_error(q_a=q_a,
q_b=q_b,
distance=distance,
robot=robot)
np.testing.assert_allclose(actual_error, 0)
# test 2D input
actual_error = compute_relative_errors(
qs_a=np.tile(q_a, (10, 1)),
qs_b=np.tile(q_b, (10, 1)),
distances=np.tile(distance, (10, 1)),
robot=robot,
)
np.testing.assert_allclose(actual_error, 0)
def test_ik(q: np.ndarray, q_offset: np.ndarray):
"""Test."""
robot = Robot.from_parameters(ur10())
pose = robot.fk(q)
# IK is hard to solve without a decent seed
q_actual = robot.ik(pose, q=robot.clamp_joints(q + q_offset))
if q_actual is None:
# ik couldn't be solved
# don't fail test
return
actual_pose = robot.fk(q_actual)
# test the matrix with lower accuracy
# rotation components are hard to achieve when x0 isn't good
np.testing.assert_allclose(actual_pose, pose, atol=1)
# test the position with higher accuracy
desired_position = pose[:-1, -1]
actual_position = actual_pose[:-1, -1]
np.testing.assert_allclose(actual_position, desired_position, atol=1e-1)
from pybotics.errors import PyboticsError
from pybotics.optimization import (
OptimizationHandler,
compute_absolute_error,
compute_absolute_errors,
optimize_accuracy,
compute_relative_error,
compute_relative_errors,
)
from pybotics.predefined_models import ur10
from pybotics.robot import Robot
@given(q=arrays(
shape=(len(ur10()), ),
dtype=float,
elements=st.floats(allow_nan=False, allow_infinity=False),
))
def test_compute_absolute_errors(q: np.ndarray):
"""Test."""
robot = Robot.from_parameters(ur10())
pose = robot.fk(q)
p = pose[:-1, -1]
# test 1D input
actual_error = compute_absolute_error(q=q, position=p, robot=robot)
np.testing.assert_allclose(actual_error, 0)
# test 2D input
actual_error = compute_absolute_errors(qs=np.tile(q, (10, 1)),
def test_home_position():
"""Test."""
robot = Robot.from_parameters(ur10())
x = np.ones(len(robot))
robot.home_position = x
np.testing.assert_allclose(robot.home_position, x)
def test_to_json():
"""Test."""
robot = Robot.from_parameters(ur10())
robot.to_json()
def test_random_joints():
"""Test."""
robot = Robot.from_parameters(ur10())
robot.random_joints()
robot.random_joints(in_place=True)