Skip to content

Commit 4df2e05

Browse files
committed
Combine tests.
1 parent f3cba8f commit 4df2e05

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tests/test_utils.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -108,10 +108,6 @@ def test_get_freejoint_dims(self):
108108
np.asarray(list(range(0, 6))),
109109
)
110110

111-
def test_get_dof_ids_throws_error_if_joint_name_invalid(self):
112-
with self.assertRaises(InvalidJointName):
113-
utils.get_dof_ids(self.model, ["invalid_joint_name"])
114-
115111
def test_get_dof_ids(self):
116112
xml_str = """
117113
<mujoco>
@@ -123,8 +119,8 @@ def test_get_dof_ids(self):
123119
<joint type="hinge" name="hinge1" range="0 1.57"/>
124120
<geom type="sphere" size=".1" mass=".1"/>
125121
<body>
126-
<joint type="hinge" name="hinge2" range="0 1.57"/>
127-
<geom type="sphere" size=".1" mass=".1"/>
122+
<joint type="hinge" name="hinge2" range="0 1.57"/>
123+
<geom type="sphere" size=".1" mass=".1"/>
128124
</body>
129125
</body>
130126
</body>
@@ -134,6 +130,8 @@ def test_get_dof_ids(self):
134130
model = mujoco.MjModel.from_xml_string(xml_str)
135131
dof_ids = utils.get_dof_ids(model, ["ball", "hinge2"])
136132
np.testing.assert_allclose(dof_ids, [0, 1, 2, 4])
133+
with self.assertRaises(InvalidJointName):
134+
utils.get_dof_ids(model, ["invalid_joint_name"])
137135

138136
def test_get_subtree_geom_ids(self):
139137
xml_str = """

0 commit comments

Comments
 (0)