Add non-tree element lists to python bindings.

Fixes #1752.

PiperOrigin-RevId: 649405039
Change-Id: Ie0b575a3ec912b773141694397ac2f394bff1227
This commit is contained in:
Alessio Quaglino
2024-07-04 07:23:05 -07:00
committed by Copybara-Service
parent d831153401
commit a0945e4b5a
10 changed files with 984 additions and 0 deletions
+13
View File
@@ -217,6 +217,19 @@ class SpecsTest(absltest.TestCase):
</mujoco>
"""))
def test_element_list(self):
spec = mujoco.MjSpec()
sensor1 = spec.add_sensor()
sensor2 = spec.add_sensor()
sensor3 = spec.add_sensor()
sensor1.name = 'sensor1'
sensor2.name = 'sensor2'
sensor3.name = 'sensor3'
self.assertLen(spec.sensors, 3)
self.assertEqual(spec.sensors[0].name, 'sensor1')
self.assertEqual(spec.sensors[1].name, 'sensor2')
self.assertEqual(spec.sensors[2].name, 'sensor3')
if __name__ == '__main__':
absltest.main()