diff --git a/tests/motors/test_dynamixel.py b/tests/motors/test_dynamixel.py index f3d7fa58..dcce8f69 100644 --- a/tests/motors/test_dynamixel.py +++ b/tests/motors/test_dynamixel.py @@ -3,14 +3,19 @@ import sys from typing import Generator from unittest.mock import MagicMock, patch -import dynamixel_sdk as dxl import pytest from lerobot.common.motors import Motor, MotorCalibration, MotorNormMode from lerobot.common.motors.dynamixel import MODEL_NUMBER_TABLE, DynamixelMotorsBus from lerobot.common.motors.dynamixel.tables import X_SERIES_CONTROL_TABLE from lerobot.common.utils.encoding_utils import encode_twos_complement -from tests.mocks.mock_dynamixel import MockMotors, MockPortHandler + +try: + import dynamixel_sdk as dxl + + from tests.mocks.mock_dynamixel import MockMotors, MockPortHandler +except (ImportError, ModuleNotFoundError): + pytest.skip("dynamixel_sdk not available", allow_module_level=True) @pytest.fixture(autouse=True) diff --git a/tests/motors/test_feetech.py b/tests/motors/test_feetech.py index 37e70c9b..337d2df7 100644 --- a/tests/motors/test_feetech.py +++ b/tests/motors/test_feetech.py @@ -4,13 +4,18 @@ from typing import Generator from unittest.mock import MagicMock, patch import pytest -import scservo_sdk as scs from lerobot.common.motors import Motor, MotorCalibration, MotorNormMode from lerobot.common.motors.feetech import MODEL_NUMBER, MODEL_NUMBER_TABLE, FeetechMotorsBus from lerobot.common.motors.feetech.tables import STS_SMS_SERIES_CONTROL_TABLE from lerobot.common.utils.encoding_utils import encode_sign_magnitude -from tests.mocks.mock_feetech import MockMotors, MockPortHandler + +try: + import scservo_sdk as scs + + from tests.mocks.mock_feetech import MockMotors, MockPortHandler +except (ImportError, ModuleNotFoundError): + pytest.skip("scservo_sdk not available", allow_module_level=True) @pytest.fixture(autouse=True)