From 0843310967b724bfb3739a9e32d14da2adb610a3 Mon Sep 17 00:00:00 2001
From: Guilhem Saurel <guilhem.saurel@laas.fr>
Date: Wed, 6 Nov 2019 16:43:24 +0100
Subject: [PATCH] load ur 3/5 with or without its gripper

---
 python/example_robot_data/robots_loader.py |  5 +++--
 unittest/test_load.py                      | 12 ++++++++++++
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/python/example_robot_data/robots_loader.py b/python/example_robot_data/robots_loader.py
index 5e51873..6374a86 100644
--- a/python/example_robot_data/robots_loader.py
+++ b/python/example_robot_data/robots_loader.py
@@ -236,8 +236,9 @@ def loadICub(reduced=True):
     return robot
 
 
-def loadUR(robot=5, limited=False):
-    URDF_FILENAME = "ur%i%s_robot.urdf" % (robot, "_joint_limited" if limited else '')
+def loadUR(robot=5, limited=False, gripper=False):
+    assert (not (gripper and (robot == 10 or limited)))
+    URDF_FILENAME = "ur%i%s_%s.urdf" % (robot, "_joint_limited" if limited else '', 'gripper' if gripper else 'robot')
     URDF_SUBPATH = "/ur_description/urdf/" + URDF_FILENAME
     modelPath = getModelPath(URDF_SUBPATH)
     return RobotWrapper.BuildFromURDF(modelPath + URDF_SUBPATH, [modelPath])
diff --git a/unittest/test_load.py b/unittest/test_load.py
index 2052199..1c5dcee 100755
--- a/unittest/test_load.py
+++ b/unittest/test_load.py
@@ -101,6 +101,18 @@ class UR5Test(RobotTestCase):
     RobotTestCase.NV = 6
 
 
+class UR5LimitedTest(RobotTestCase):
+    RobotTestCase.ROBOT = example_robot_data.loadUR(limited=True)
+    RobotTestCase.NQ = 6
+    RobotTestCase.NV = 6
+
+
+class UR5GripperTest(RobotTestCase):
+    RobotTestCase.ROBOT = example_robot_data.loadUR(gripper=True)
+    RobotTestCase.NQ = 6
+    RobotTestCase.NV = 6
+
+
 class KinovaTest(RobotTestCase):
     RobotTestCase.ROBOT = example_robot_data.loadKinova()
     RobotTestCase.NQ = 9
-- 
GitLab