From e499d607421500a61d608127d1a392e0ebfea3b0 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Fri, 27 Sep 2024 12:29:58 +0200 Subject: [PATCH] fix unit test --- tests/test_control_robot.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py index 75238004..407a3290 100644 --- a/tests/test_control_robot.py +++ b/tests/test_control_robot.py @@ -37,6 +37,9 @@ from tests.utils import DEFAULT_CONFIG_PATH, DEVICE, TEST_ROBOT_TYPES, require_r @pytest.mark.parametrize("robot_type, mock", TEST_ROBOT_TYPES) @require_robot def test_teleoperate(request, robot_type, mock): + if mock: + request.getfixturevalue("patch_builtins_input") + robot = make_robot(robot_type, mock=mock) teleoperate(robot, teleop_time_s=1) teleoperate(robot, fps=30, teleop_time_s=1) @@ -85,6 +88,9 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock): if mock: request.getfixturevalue("patch_builtins_input") + if robot_type == "aloha": + pytest.skip("TODO(rcadene): enable test once aloha_real and act_aloha_real are merged") + env_name = "koch_real" policy_name = "act_koch_real"