chao.wu/Lerobot
1
0
Fork 0
Code Pull Requests Activity

Fix test_teleoperate (#1216)

Browse Source
This commit is contained in:
Simon Alibert 2025-06-06 09:38:37 +02:00 committed by GitHub
parent a28f02ecb3
commit 9e6f49f507
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_control_robot.py
View File

@ -1,5 +1,3 @@
import time
from lerobot.calibrate import CalibrateConfig, calibrate from lerobot.calibrate import CalibrateConfig, calibrate
from lerobot.record import DatasetRecordConfig, RecordConfig, record from lerobot.record import DatasetRecordConfig, RecordConfig, record
from lerobot.replay import DatasetReplayConfig, ReplayConfig, replay from lerobot.replay import DatasetReplayConfig, ReplayConfig, replay
@ -18,17 +16,12 @@ def test_calibrate():
def test_teleoperate(): def test_teleoperate():
robot_cfg = MockRobotConfig() robot_cfg = MockRobotConfig()
teleop_cfg = MockTeleopConfig() teleop_cfg = MockTeleopConfig()
expected_duration = 0.1
cfg = TeleoperateConfig( cfg = TeleoperateConfig(
robot=robot_cfg, robot=robot_cfg,
teleop=teleop_cfg, teleop=teleop_cfg,
teleop_time_s=expected_duration, teleop_time_s=0.1,
) )
start = time.perf_counter()
teleoperate(cfg) teleoperate(cfg)
actual_duration = time.perf_counter() - start
assert actual_duration <= expected_duration * 1.1
def test_record_and_resume(tmp_path): def test_record_and_resume(tmp_path):