[introspect] Extract struct array extents from comments.

PiperOrigin-RevId: 687254268 Change-Id: Ia70dca073eaa4a99c60f76b7c69c730856193512
2024-10-18 04:34:28 -07:00
parent 43a7493d17
commit 62c0ef4262
4 changed files with 1024 additions and 503 deletions
@@ -40,7 +40,8 @@ class StructsTest(absltest.TestCase):
        self.assertEqual(field.doc, 'warning statistics')
      elif field.name == 'qpos':
        self.assertEqual(field.type, type_parsing.parse_type('mjtNum*'))
-        self.assertEqual(re.sub(r'\s+', ' ', field.doc), 'position (nq x 1)')
+        self.assertEqual(field.doc, 'position')
+        self.assertEqual(field.array_extent, ('nq',))

    self.assertIn('warning', field_names)
    self.assertIn('qpos', field_names)