fix(sim): split transfer and air-insert camera sets

This commit is contained in:
Logic
2026-05-03 10:14:04 +08:00
parent b1116e489f
commit bc4caf355b
7 changed files with 26 additions and 36 deletions

View File

@@ -94,10 +94,10 @@ class AirInsertTaskRegistrationTest(unittest.TestCase):
is_render=False,
control_freq=30,
is_interpolate=True,
cam_view="left_side",
cam_view="front",
)
def test_diana_table_scene_uses_left_side_camera_instead_of_angle(self):
def test_diana_table_scene_exposes_only_top_and_front_scene_cameras(self):
xml_path = (
pathlib.Path(__file__).resolve().parents[1]
/ "roboimi/assets/models/manipulators/DianaMed/table_square.xml"
@@ -106,11 +106,11 @@ class AirInsertTaskRegistrationTest(unittest.TestCase):
cameras = {camera.attrib["name"]: camera.attrib for camera in root.findall(".//camera")}
self.assertNotIn("angle", cameras, "DianaMed scene should stop exposing the old angle camera")
self.assertIn("left_side", cameras, "DianaMed scene should expose the left-side task camera")
left_side_pos = np.fromstring(cameras["left_side"]["pos"], sep=" ")
self.assertLess(float(left_side_pos[0]), 0.0)
self.assertEqual(cameras["left_side"].get("mode"), "targetbody")
self.assertEqual(cameras["left_side"].get("target"), "table")
self.assertNotIn("left_side", cameras, "DianaMed scene should no longer expose left_side")
self.assertIn("top", cameras)
self.assertIn("front", cameras)
self.assertEqual(cameras["top"].get("mode"), "targetbody")
self.assertEqual(cameras["top"].get("target"), "table")
class AirInsertResetAndStateHelpersTest(unittest.TestCase):