mirror of
https://github.com/AtsushiSakai/PythonRobotics.git
synced 2026-02-18 01:11:33 -05:00
mypy fix test
This commit is contained in:
@@ -3,9 +3,10 @@ Forward Kinematics for an n-link arm in 3D
|
||||
Author: Takayuki Murooka (takayuki5168)
|
||||
"""
|
||||
import math
|
||||
from NLinkArm import NLinkArm
|
||||
from NLinkArm3d import NLinkArm
|
||||
import random
|
||||
|
||||
|
||||
def random_val(min_val, max_val):
|
||||
return min_val + random.random() * (max_val - min_val)
|
||||
|
||||
@@ -14,17 +15,17 @@ if __name__ == "__main__":
|
||||
print("Start solving Forward Kinematics 10 times")
|
||||
|
||||
# init NLinkArm with Denavit-Hartenberg parameters of PR2
|
||||
n_link_arm = NLinkArm([[0., -math.pi/2, .1, 0.],
|
||||
[math.pi/2, math.pi/2, 0., 0.],
|
||||
[0., -math.pi/2, 0., .4],
|
||||
[0., math.pi/2, 0., 0.],
|
||||
[0., -math.pi/2, 0., .321],
|
||||
[0., math.pi/2, 0., 0.],
|
||||
n_link_arm = NLinkArm([[0., -math.pi / 2, .1, 0.],
|
||||
[math.pi / 2, math.pi / 2, 0., 0.],
|
||||
[0., -math.pi / 2, 0., .4],
|
||||
[0., math.pi / 2, 0., 0.],
|
||||
[0., -math.pi / 2, 0., .321],
|
||||
[0., math.pi / 2, 0., 0.],
|
||||
[0., 0., 0., 0.]])
|
||||
|
||||
# execute FK 10 times
|
||||
for i in range(10):
|
||||
n_link_arm.set_joint_angles([random_val(-1, 1) for j in range(len(n_link_arm.link_list))])
|
||||
|
||||
ee_pose = n_link_arm.forward_kinematics(plot=True)
|
||||
n_link_arm.set_joint_angles(
|
||||
[random_val(-1, 1) for j in range(len(n_link_arm.link_list))])
|
||||
|
||||
ee_pose = n_link_arm.forward_kinematics(plot=True)
|
||||
|
||||
Reference in New Issue
Block a user