Fix lint errors

tracetools_test/test/test_node.py

@@ -50,7 +50,7 @@ class TestNode(TraceTestCase):
             self.assertValidHandle(event, ['node_handle', 'rmw_handle'])
             self.assertStringFieldNotEmpty(event, 'node_name')
             self.assertStringFieldNotEmpty(event, 'namespace')
-        
+
         # Check that the launched nodes have a corresponding rcl_node_init event
         node_name_fields = [self.get_field(e, 'node_name') for e in rcl_node_init_events]
         for node_name in self._nodes:

tracetools_test/test/test_service_callback.py

@@ -54,8 +54,14 @@ class TestServiceCallback(TraceTestCase):
         for node in self._nodes:
             test_start_events = self.get_events_with_procname(node, start_events)
             test_end_events = self.get_events_with_procname(node, end_events)
-            self.assertGreater(len(test_start_events), 0, f'no start_callback events for node: {node}')
+            self.assertGreater(
-            self.assertGreater(len(test_end_events), 0, f'no end_callback events for node: {node}')
+                len(test_start_events),
+                0,
+                f'no start_callback events for node: {node}')
+            self.assertGreater(
+                len(test_end_events),
+                0,
+                f'no end_callback events for node: {node}')
 
 
 if __name__ == '__main__':

tracetools_test/test/test_subscription.py

@@ -44,7 +44,8 @@ class TestSubscription(TraceTestCase):
             self.assertValidQueueDepth(event, 'queue_depth')
             self.assertStringFieldNotEmpty(event, 'topic_name')
 
-        callback_added_events = self.get_events_with_name('ros2:rclcpp_subscription_callback_added')
+        callback_added_events = self.get_events_with_name(
+            'ros2:rclcpp_subscription_callback_added')
         for event in callback_added_events:
             self.assertValidHandle(event, ['subscription_handle', 'callback'])
 

tracetools_test/tracetools_test/case.py

@@ -16,12 +16,11 @@
 
 import time
 from typing import Any
-from typing import Dict
 from typing import List
 from typing import Union
 import unittest
 
-# from .utils import cleanup_trace
+from .utils import cleanup_trace
 from .utils import DictEvent
 from .utils import get_event_name
 from .utils import get_event_names
@@ -98,8 +97,7 @@ class TraceTestCase(unittest.TestCase):
         self.assertProcessNamesExist(self._nodes)
 
     def tearDown(self):
-        pass
+        cleanup_trace(self._full_path)
-        # cleanup_trace(self._full_path)
 
     def assertEventsOrderSet(self, event_names: List[str]):
         """
@@ -182,7 +180,7 @@ class TraceTestCase(unittest.TestCase):
         if matching_event_name is not None:
             events = self.get_events_with_name(matching_event_name, events)
         field_value = self.get_field(initial_event, field_name)
-        
+
         # Get events with that handle
         matches = self.get_events_with_field_value(
             field_name,
@@ -202,7 +200,6 @@ class TraceTestCase(unittest.TestCase):
             1,
             'matching field event not after initial event')
 
-
     def assertFieldEquals(self, event: DictEvent, field_name: str, value: Any):
         """
         Check the value of a field.
@@ -229,7 +226,7 @@ class TraceTestCase(unittest.TestCase):
             self.fail(str(e))
         else:
             return value
-        
+
     def get_procname(self, event: DictEvent) -> str:
         """
         Get procname.
@@ -270,7 +267,7 @@ class TraceTestCase(unittest.TestCase):
         if events is None:
             events = self._events
         return [e for e in events if self.get_procname(e) == procname[:15]]
-    
+
     def get_events_with_field_value(
         self,
         field_name: str,
@@ -278,7 +275,7 @@ class TraceTestCase(unittest.TestCase):
         events: List[DictEvent] = None
     ) -> List[DictEvent]:
         """
-        Get all events with the given field:value
+        Get all events with the given field:value.
 
         :param field_name: the name of the field to check
         :param field_value: the value of the field to check

tracetools_test/tracetools_test/utils.py

@@ -88,8 +88,10 @@ def cleanup_trace(full_path: str) -> None:
     """
     shutil.rmtree(full_path)
 
+
 DictEvent = Dict[str, Any]
 
+
 def get_trace_events(trace_directory: str) -> List[DictEvent]:
     """
     Get the events of a trace.