wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Assert ping stub called

This commit is contained in:
Simon Alibert 2025-03-22 18:53:57 +01:00
parent fedac994c3
commit c2e761437d
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/motors/test_dynamixel.py
View File

@ -101,7 +101,7 @@ def test_abc_implementation(dummy_motors):
], ],
) )
def test_ping(idx, model_nb, mock_motors, dummy_motors): def test_ping(idx, model_nb, mock_motors, dummy_motors):
mock_motors.build_ping_stub(idx, model_nb) stub_name = mock_motors.build_ping_stub(idx, model_nb)
motors_bus = DynamixelMotorsBus( motors_bus = DynamixelMotorsBus(
port=mock_motors.port, port=mock_motors.port,
motors=dummy_motors, motors=dummy_motors,
@ -111,6 +111,7 @@ def test_ping(idx, model_nb, mock_motors, dummy_motors):
ping_model_nb = motors_bus.ping(idx) ping_model_nb = motors_bus.ping(idx)
assert ping_model_nb == model_nb assert ping_model_nb == model_nb
assert mock_motors.stubs[stub_name].called
def test_broadcast_ping(mock_motors, dummy_motors): def test_broadcast_ping(mock_motors, dummy_motors):
@ -119,7 +120,7 @@ def test_broadcast_ping(mock_motors, dummy_motors):
2: [1120, 30], 2: [1120, 30],
3: [1190, 10], 3: [1190, 10],
} }
mock_motors.build_broadcast_ping_stub(expected_pings) stub_name = mock_motors.build_broadcast_ping_stub(expected_pings)
motors_bus = DynamixelMotorsBus( motors_bus = DynamixelMotorsBus(
port=mock_motors.port, port=mock_motors.port,
motors=dummy_motors, motors=dummy_motors,
@ -129,6 +130,7 @@ def test_broadcast_ping(mock_motors, dummy_motors):
ping_list = motors_bus.broadcast_ping() ping_list = motors_bus.broadcast_ping()
assert ping_list == expected_pings assert ping_list == expected_pings
assert mock_motors.stubs[stub_name].called
@pytest.mark.parametrize( @pytest.mark.parametrize(

6
tests/motors/test_feetech.py
View File

@ -102,7 +102,7 @@ def test_abc_implementation(dummy_motors):
], ],
) )
def test_ping(idx, model_nb, mock_motors, dummy_motors): def test_ping(idx, model_nb, mock_motors, dummy_motors):
mock_motors.build_ping_stub(idx, model_nb) stub_name = mock_motors.build_ping_stub(idx, model_nb)
motors_bus = FeetechMotorsBus( motors_bus = FeetechMotorsBus(
port=mock_motors.port, port=mock_motors.port,
motors=dummy_motors, motors=dummy_motors,
@ -112,6 +112,7 @@ def test_ping(idx, model_nb, mock_motors, dummy_motors):
ping_model_nb = motors_bus.ping(idx) ping_model_nb = motors_bus.ping(idx)
assert ping_model_nb == model_nb assert ping_model_nb == model_nb
assert mock_motors.stubs[stub_name].called
@pytest.mark.skip("TODO") @pytest.mark.skip("TODO")
@ -121,7 +122,7 @@ def test_broadcast_ping(mock_motors, dummy_motors):
2: [1120, 30], 2: [1120, 30],
3: [1190, 10], 3: [1190, 10],
} }
mock_motors.build_broadcast_ping_stub(expected_pings) stub_name = mock_motors.build_broadcast_ping_stub(expected_pings)
motors_bus = FeetechMotorsBus( motors_bus = FeetechMotorsBus(
port=mock_motors.port, port=mock_motors.port,
motors=dummy_motors, motors=dummy_motors,
@ -131,6 +132,7 @@ def test_broadcast_ping(mock_motors, dummy_motors):
ping_list = motors_bus.broadcast_ping() ping_list = motors_bus.broadcast_ping()
assert ping_list == expected_pings assert ping_list == expected_pings
assert mock_motors.stubs[stub_name].called
@pytest.mark.parametrize( @pytest.mark.parametrize(