fix unit tests (minimal)
This commit is contained in:
@@ -3,7 +3,6 @@ import time
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from lerobot.common.robot_devices.motors.dynamixel import DynamixelMotorsBus, find_port
|
|
||||||
from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
|
from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
|
||||||
from tests.utils import require_koch
|
from tests.utils import require_koch
|
||||||
|
|
||||||
@@ -14,6 +13,7 @@ def test_motors_bus(request):
|
|||||||
# TODO(rcadene): test logs
|
# TODO(rcadene): test logs
|
||||||
# TODO(rcadene): test calibration
|
# TODO(rcadene): test calibration
|
||||||
# TODO(rcadene): add compatibility with other motors bus
|
# TODO(rcadene): add compatibility with other motors bus
|
||||||
|
from lerobot.common.robot_devices.motors.dynamixel import DynamixelMotorsBus
|
||||||
|
|
||||||
# Test instantiating a common motors structure.
|
# Test instantiating a common motors structure.
|
||||||
# Here the one from Alexander Koch follower arm.
|
# Here the one from Alexander Koch follower arm.
|
||||||
@@ -87,4 +87,6 @@ def test_motors_bus(request):
|
|||||||
|
|
||||||
@require_koch
|
@require_koch
|
||||||
def test_find_port(request):
|
def test_find_port(request):
|
||||||
|
from lerobot.common.robot_devices.motors.dynamixel import find_port
|
||||||
|
|
||||||
find_port()
|
find_port()
|
||||||
|
|||||||
@@ -5,7 +5,6 @@ import pytest
|
|||||||
import torch
|
import torch
|
||||||
|
|
||||||
from lerobot.common.robot_devices.robots.factory import make_robot
|
from lerobot.common.robot_devices.robots.factory import make_robot
|
||||||
from lerobot.common.robot_devices.robots.koch import KochRobot
|
|
||||||
from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
|
from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
|
||||||
from tests.utils import require_koch
|
from tests.utils import require_koch
|
||||||
|
|
||||||
@@ -15,6 +14,7 @@ def test_robot(tmpdir, request):
|
|||||||
# TODO(rcadene): measure fps in nightly?
|
# TODO(rcadene): measure fps in nightly?
|
||||||
# TODO(rcadene): test logs
|
# TODO(rcadene): test logs
|
||||||
# TODO(rcadene): add compatibility with other robots
|
# TODO(rcadene): add compatibility with other robots
|
||||||
|
from lerobot.common.robot_devices.robots.koch import KochRobot
|
||||||
|
|
||||||
# Save calibration preset
|
# Save calibration preset
|
||||||
calibration = {
|
calibration = {
|
||||||
|
|||||||
Reference in New Issue
Block a user