35
35
from rclpy .node import Node
36
36
from rclpy .qos import QoSProfile
37
37
38
- from rai_sim .launch_manager import Ros2LaunchManager
38
+ from rai_sim .launch_manager import ROS2LaunchManager
39
39
from rai_sim .o3de .o3de_bridge import O3DExROS2Bridge , O3DExROS2SimulationConfig
40
40
from rai_sim .simulation_bridge import Entity , SceneConfig , SpawnedEntity
41
41
@@ -69,12 +69,12 @@ def test_load_scene_config(sample_base_yaml_config: Path):
69
69
70
70
71
71
class TestO3DExROS2Bridge (unittest .TestCase ):
72
- @patch ("rai_sim.o3de.o3de_bridge.Ros2LaunchManager " )
72
+ @patch ("rai_sim.o3de.o3de_bridge.ROS2LaunchManager " )
73
73
def setUp (self , mock_launch_manager_class ):
74
74
self .mock_connector = MagicMock (spec = ROS2Connector )
75
75
self .mock_logger = MagicMock ()
76
76
77
- self .mock_launch_manager = MagicMock (spec = Ros2LaunchManager )
77
+ self .mock_launch_manager = MagicMock (spec = ROS2LaunchManager )
78
78
mock_launch_manager_class .return_value = self .mock_launch_manager
79
79
80
80
self .bridge = O3DExROS2Bridge (
@@ -327,9 +327,9 @@ def test_get_transform_signature(self):
327
327
"timeout_sec" : float ,
328
328
}
329
329
330
- assert list (parameters .keys ()) == list (expected_params . keys ()), (
331
- f"Parameter names do not match, expected: { list ( expected_params .keys ()) } , got: { list ( parameters . keys ()) } "
332
- )
330
+ assert list (parameters .keys ()) == list (
331
+ expected_params .keys ()
332
+ ), f"Parameter names do not match, expected: { list ( expected_params . keys ()) } , got: { list ( parameters . keys ()) } "
333
333
334
334
for param_name , expected_type in expected_params .items ():
335
335
param = parameters [param_name ]
@@ -340,9 +340,9 @@ def test_get_transform_signature(self):
340
340
)
341
341
342
342
# Check return type explicitly
343
- assert signature . return_annotation is ROS2TransformStamped , (
344
- f"Return type is incorrect, expected: TransformStamped, got: { signature .return_annotation } "
345
- )
343
+ assert (
344
+ signature .return_annotation is ROS2TransformStamped
345
+ ), f"Return type is incorrect, expected: TransformStamped, got: { signature . return_annotation } "
346
346
347
347
def test_send_message_signature (self ):
348
348
signature = inspect .signature (self .connector .send_message )
@@ -394,9 +394,9 @@ def test_get_topics_names_and_types_signature(self):
394
394
395
395
expected_params : dict [str , type ] = {}
396
396
397
- assert list (parameters .keys ()) == list (expected_params . keys ()), (
398
- f"Parameter names do not match, expected: { list ( expected_params .keys ()) } , got: { list ( parameters . keys ()) } "
399
- )
397
+ assert list (parameters .keys ()) == list (
398
+ expected_params .keys ()
399
+ ), f"Parameter names do not match, expected: { list ( expected_params . keys ()) } , got: { list ( parameters . keys ()) } "
400
400
401
401
for param_name , expected_type in expected_params .items ():
402
402
param = parameters [param_name ]
0 commit comments