# 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):
