Fix unit tests

2024-10-13 18:31:34 +02:00
parent d02e204e10
commit eed7b55fe3
6 changed files with 94 additions and 38 deletions
--- a/tests/test_robots.py
+++ b/tests/test_robots.py
@@ -127,6 +127,7 @@ def test_robot(tmpdir, request, robot_type, mock):
            # TODO(rcadene): skipping image for now as it's challenging to assess equality between two consecutive frames
            continue
        assert torch.allclose(captured_observation[name], observation[name], atol=1)
+        assert captured_observation[name].shape == observation[name].shape

    # Test send_action can run
    robot.send_action(action["action"])