Fix assigning bytes to texture data.

PiperOrigin-RevId: 706747323 Change-Id: I6b0ee67d4f35db3a149ba8d9979546d3c712bb8b
2024-12-16 10:04:06 -08:00
parent a7eb6efd4e
commit 644dfc7fb5
2 changed files with 10 additions and 3 deletions
@@ -891,6 +891,12 @@ class SpecsTest(absltest.TestCase):
    with self.assertRaises(IndexError):
      material.textures[-1] = 'x'

+  def test_assign_texture(self):
+    spec = mujoco.MjSpec()
+    texture = spec.add_texture(name='texture', height=2, width=2)
+    texture.data = np.zeros((2, 2, 3), dtype=np.uint8).tobytes()
+    spec.compile()
+
  def test_attach_units(self):
    child = mujoco.MjSpec()
    parent = mujoco.MjSpec()