bindings_std_map.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):
        map = pin.StdMap_String_VectorXd()
        keys = []
        for k in range(100):
            key_name = "key_" + str(k + 1)
            keys.append(key_name)
            map[key_name] = np.random.rand((10))
 
        pickle.dump(map, open("save_std_map.p", "wb"))
 
        map_loaded = pickle.load(open("save_std_map.p", "rb"))
        for key in keys:
            self.assertApprox(map[key], map_loaded[key])
 
 
if __name__ == "__main__":
    unittest.main()