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

Add pytest param ids

This commit is contained in:
Simon Alibert 2025-03-20 09:44:47 +01:00
parent 2c1bb766ff
commit dd1f33e5ed
1 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/motors/test_dynamixel.py
View File

@ -26,15 +26,26 @@ def test_autouse_patch():
@pytest.mark.parametrize( @pytest.mark.parametrize(
"value, n_bytes, expected", "value, n_bytes, expected",
[ [
(0x12, 1, [0x12]), # Single byte (0x12, 1, [0x12]),
(0x1234, 2, [0x34, 0x12]), # Two bytes (0x1234, 2, [0x34, 0x12]),
(0x12345678, 4, [0x78, 0x56, 0x34, 0x12]), # Four bytes (0x12345678, 4, [0x78, 0x56, 0x34, 0x12]),
(0, 1, [0x00]), # Zero with 1 byte (0, 1, [0x00]),
(0, 2, [0x00, 0x00]), # Zero with 2 bytes (0, 2, [0x00, 0x00]),
(0, 4, [0x00, 0x00, 0x00, 0x00]), # Zero with 4 bytes (0, 4, [0x00, 0x00, 0x00, 0x00]),
(255, 1, [0xFF]), # Max single byte (255, 1, [0xFF]),
(65535, 2, [0xFF, 0xFF]), # Max two bytes (65535, 2, [0xFF, 0xFF]),
(4294967295, 4, [0xFF, 0xFF, 0xFF, 0xFF]), # Max four bytes (4294967295, 4, [0xFF, 0xFF, 0xFF, 0xFF]),
],
ids=[
"1 byte",
"2 bytes",
"4 bytes",
"0 with 1 byte",
"0 with 2 bytes",
"0 with 4 bytes",
"max single byte",
"max two bytes",
"max four bytes",
], ],
) # fmt: skip ) # fmt: skip
def test_split_int_bytes(value, n_bytes, expected): def test_split_int_bytes(value, n_bytes, expected):
@ -58,7 +69,7 @@ def test_split_int_bytes_large_number():
def test_abc_implementation(): def test_abc_implementation():
# Instantiation should raise an error if the class doesn't implements abstract methods/properties """Instantiation should raise an error if the class doesn't implement abstract methods/properties."""
DynamixelMotorsBus(port="/dev/dummy-port", motors={"dummy": (1, "xl330-m077")}) DynamixelMotorsBus(port="/dev/dummy-port", motors={"dummy": (1, "xl330-m077")})
@ -70,6 +81,7 @@ def test_abc_implementation():
["dummy_1", "dummy_2", "dummy_3"], ["dummy_1", "dummy_2", "dummy_3"],
[1, "dummy_2", 3], [1, "dummy_2", 3],
], ],
ids=["None", "by ids", "by names", "mixed"],
) )
def test_read_all_motors(motors): def test_read_all_motors(motors):
mock_motors = MockMotors([1, 2, 3]) mock_motors = MockMotors([1, 2, 3])