From 20129cd4c28fd9a2632fb1c608530084b7425ebb Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Sun, 11 May 2025 12:52:17 +0200 Subject: [PATCH] Fix tests (no-extras install) --- tests/motors/test_dynamixel.py | 9 +++++++-- tests/motors/test_feetech.py | 9 +++++++-- 2 files changed, 14 insertions(+), 4 deletions(-) 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)