Source code for robots.pr2.head_sockets
#! /usr/bin/env python
"""
This script tests the PR2 torso armature joint
"""
from morse.testing.testing import MorseTestCase
import logging; logger = logging.getLogger("morsetesting.general")
# Include this import to be able to use your test file as a regular
# builder script, ie, usable with: 'morse [run|exec] base_testing.py
try:
from morse.builder import *
except ImportError:
pass
import sys
import time
from pymorse import Morse
[docs]def getjoint(name):
with Morse() as simu:
joints = simu.pr2.posture.get()
return joints[name]
[docs]class PR2TorsoTest(MorseTestCase):
[docs] def setUpEnv(self):
from morse.builder.robots import BasePR2
pr2 = BasePR2()
pr2.add_stream('socket')
env = Environment('empty', fastmode=True)
env.set_camera_rotation([1.0470, 0, 0.7854])
[docs] def test_head_pan(self):
joint = "head_pan"
self.assertAlmostEquals(getjoint(joint), 0.0, 3)
with Morse() as simu:
logger.info("Paning the head 1")
action = simu.pr2.head.rotate(joint, 1, 1).result() # speed = 1 rad/s
self.assertAlmostEquals(getjoint(joint), 1, 1)
########################## Run these tests ##########################
if __name__ == "__main__":
from morse.testing.testing import main
main(PR2TorsoTest)