diff --git a/tests/motors/test_dynamixel.py b/tests/motors/test_dynamixel.py index ba6d75ba..22cdef23 100644 --- a/tests/motors/test_dynamixel.py +++ b/tests/motors/test_dynamixel.py @@ -3,6 +3,7 @@ from unittest.mock import patch import pytest +from lerobot.common.motors.dynamixel import DynamixelMotorsBus from tests.mocks import mock_dynamixel_sdk @@ -18,15 +19,11 @@ def test_patch_sdk(): def test_abc_implementation(): - from lerobot.common.motors.dynamixel import DynamixelMotorsBus - # Instantiation should raise an error if the class doesn't implements abstract methods/properties DynamixelMotorsBus(port="/dev/dummy-port", motors={"dummy": (1, "xl330-m077")}) def test_configure_motors_all_ids_1(): - from lerobot.common.motors.dynamixel import DynamixelMotorsBus - # see X_SERIES_BAUDRATE_TABLE smaller_baudrate = 9_600 smaller_baudrate_value = 0 diff --git a/tests/motors/test_feetech.py b/tests/motors/test_feetech.py index 9805de84..6c580f6a 100644 --- a/tests/motors/test_feetech.py +++ b/tests/motors/test_feetech.py @@ -3,6 +3,7 @@ from unittest.mock import patch import pytest +from lerobot.common.motors.feetech import FeetechMotorsBus from tests.mocks import mock_scservo_sdk @@ -18,15 +19,11 @@ def test_patch_sdk(): def test_abc_implementation(): - from lerobot.common.motors.feetech import FeetechMotorsBus - # Instantiation should raise an error if the class doesn't implements abstract methods/properties FeetechMotorsBus(port="/dev/dummy-port", motors={"dummy": (1, "sts3215")}) def test_configure_motors_all_ids_1(): - from lerobot.common.motors.feetech import FeetechMotorsBus - # see SCS_SERIES_BAUDRATE_TABLE smaller_baudrate = 19_200 smaller_baudrate_value = 7