bindings_std_vector.py

Go to the documentation of this file.

import unittest
from test_case import PinocchioTestCase as TestCase

import pinocchio as pin
import numpy as np

import pickle

class TestStdMap(TestCase):

    def setUp(self):
        pass

    def test_pickle(self):
        vec = pin.StdVec_Vector3()
        for k in range(100):
            vec.append(np.random.rand((3)))

        pickle.dump( vec, open( "save_std_vec.p", "wb" ) )

        vec_loaded = pickle.load( open( "save_std_vec.p", "rb" ) )
        for k in range(len(vec)):
            self.assertApprox(vec[k],vec_loaded[k])

if __name__ == '__main__':
    unittest.main()