bindings_model.py

Go to the documentation of this file.

import unittest
import pinocchio as pin
from pinocchio.utils import np, zero
 
from test_case import PinocchioTestCase as TestCase
 
 
class TestModel(TestCase):
    def setUp(self):
        self.model = pin.buildSampleModelHumanoidRandom()
 
    def test_empty_model_sizes(self):
        model = pin.Model()
        self.assertEqual(model.nbodies, 1)
        self.assertEqual(model.nq, 0)
        self.assertEqual(model.nv, 0)
        model.name = "empty_model"
 
    def test_add_joint(self):
        model = pin.Model()
        idx = 0
        idx = model.addJoint(
            idx, pin.JointModelRY(), pin.SE3.Identity(), "joint_" + str(idx + 1)
        )
 
        MAX_EFF = 100.0
        MAX_VEL = 10.0
        MIN_POS = -1.0
        MAX_POS = 1.0
 
        me = np.array([MAX_EFF])
        mv = np.array([MAX_VEL])
        lb = np.array([MIN_POS])
        ub = np.array([MAX_POS])
        idx = model.addJoint(
            idx,
            pin.JointModelRY(),
            pin.SE3.Identity(),
            "joint_" + str(idx + 1),
            me,
            mv,
            lb,
            ub,
        )
 
        self.assertEqual(model.nbodies, 1)
        self.assertEqual(model.njoints, 3)
        self.assertEqual(model.nq, 2)
        self.assertEqual(model.nv, 2)
 
        self.assertEqual(float(model.effortLimit[1]), MAX_EFF)
        self.assertEqual(float(model.velocityLimit[1]), MAX_VEL)
        self.assertEqual(float(model.lowerPositionLimit[1]), MIN_POS)
        self.assertEqual(float(model.upperPositionLimit[1]), MAX_POS)
 
    def test_model(self):
        model = self.model
        nb = 28  # We should have 28 bodies, thus 27 joints, one of them a free-flyer.
        self.assertEqual(model.nbodies, nb)
        self.assertEqual(model.nq, nb - 1 + 6)
        self.assertEqual(model.nv, nb - 1 + 5)
 
    def test_inertias(self):
        model = self.model
        model.inertias[1] = model.inertias[2]
        self.assertApprox(model.inertias[1].np, model.inertias[2].np)
 
    def test_placements(self):
        model = self.model
        model.jointPlacements[1] = model.jointPlacements[2]
        self.assertApprox(model.jointPlacements[1].np, model.jointPlacements[2].np)
        self.assertEqual(model.parents[0], 0)
        self.assertEqual(model.parents[1], 0)
        model.parents[2] = model.parents[1]
        self.assertEqual(model.parents[2], model.parents[1])
        self.assertEqual(model.names[0], "universe")
 
    def test_gravity(self):
        self.assertApprox(self.model.gravity.np, np.array([0, 0, -9.81, 0, 0, 0]).T)
 
    def test_rnea(self):
        model = self.model
        data = model.createData()
 
        q = zero(model.nq)
        qdot = zero(model.nv)
        qddot = zero(model.nv)
        for i in range(model.nbodies):
            data.a[i] = pin.Motion.Zero()
 
        pin.rnea(model, data, q, qdot, qddot)
        for i in range(model.nbodies):
            self.assertApprox(data.v[i].np, zero(6))
        self.assertApprox(data.a_gf[0].np, -model.gravity.np)
        self.assertApprox(data.f[-1], model.inertias[-1] * data.a_gf[-1])
 
    def test_std_map_fields(self):
        model = self.model
        model.referenceConfigurations["neutral"] = pin.neutral(model)
        q_neutral = model.referenceConfigurations["neutral"]
 
        q_neutral.fill(1.0)
        self.assertApprox(model.referenceConfigurations["neutral"], q_neutral)
 
    def test_pickle(self):
        import pickle
 
        model = self.model
        filename = "model.pickle"
        with open(filename, "wb") as f:
            pickle.dump(model, f)
 
        with open(filename, "rb") as f:
            model_copy = pickle.load(f)
 
        self.assertTrue(model == model_copy)
 
 
if __name__ == "__main__":
    unittest.main()