# Internal dependencies. from absl.testing import absltest from absl.testing import parameterized from dm_control import render from dm_control.mujoco.testing import assets from dm_control.mujoco.wrapper import core from dm_control.mujoco.wrapper import mjbindings from dm_control.mujoco.wrapper.mjbindings import enums import mock import numpy as np from six.moves import cPickle from six.moves import range mjlib = mjbindings.mjlib HUMANOID_XML_PATH = assets.get_path("humanoid.xml") MODEL_WITH_ASSETS = assets.get_contents("model_with_assets.xml") ASSETS = { "texture.png": assets.get_contents("deepmind.png"), "mesh.stl": assets.get_contents("cube.stl"), "included.xml": assets.get_contents("sphere.xml") } SCALAR_TYPES = (int, float) ARRAY_TYPES = (np.ndarray, ) OUT_DIR = absltest.get_default_test_tmpdir() if not os.path.exists(OUT_DIR): os.makedirs(OUT_DIR) # Ensure that the output directory exists.
from absl.testing import parameterized from dm_control.mujoco import engine from dm_control.mujoco import wrapper from dm_control.mujoco.testing import assets from dm_control.mujoco.wrapper import mjbindings from dm_control.mujoco.wrapper.mjbindings import enums from dm_control.rl import control import mock import numpy as np import six from six.moves import cPickle from six.moves import range mjlib = mjbindings.mjlib MODEL_PATH = assets.get_path('cartpole.xml') MODEL_WITH_ASSETS = assets.get_contents('model_with_assets.xml') ASSETS = { 'texture.png': assets.get_contents('deepmind.png'), 'mesh.stl': assets.get_contents('cube.stl'), 'included.xml': assets.get_contents('sphere.xml') } class MujocoEngineTest(parameterized.TestCase): def setUp(self): super(MujocoEngineTest, self).setUp() self._physics = engine.Physics.from_xml_path(MODEL_PATH) def _assert_attributes_equal(self, actual_obj, expected_obj, attr_to_compare):