niklas/ros2_tracing
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add mypy tests through ament_mypy

Browse source 
Signed-off-by: Christophe Bedard <bedard.christophe@gmail.com>
This commit is contained in:
Christophe Bedard 2020-03-01 11:50:31 -05:00
parent aa59950fe2
commit 9450bfeea9
15 changed files with 117 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tracetools_read/tracetools_read/trace.py
View file

@ -23,9 +23,6 @@ import babeltrace
from . import DictEvent
BabeltraceEvent = babeltrace.babeltrace.Event
def is_trace_directory(path: str) -> bool:
"""
Check recursively if a path is a trace directory.
@ -42,7 +39,7 @@ def is_trace_directory(path: str) -> bool:
return traces is not None and len(traces) > 0
def get_trace_ctf_events(trace_directory: str) -> Iterable[BabeltraceEvent]:
def get_trace_ctf_events(trace_directory: str) -> Iterable[babeltrace.babeltrace.Event]:
"""
Get the events of a trace.
@ -61,7 +58,10 @@ def get_trace_events(trace_directory: str) -> List[DictEvent]:
:param trace_directory: the path to the main/top trace directory
:return: events
"""
return [event_to_dict(event) for event in get_trace_ctf_events(trace_directory)]
events: List[DictEvent] = [
event_to_dict(event) for event in get_trace_ctf_events(trace_directory)
]
return events
# List of ignored CTF fields
@ -82,7 +82,7 @@ _IGNORED_FIELDS = [
_DISCARD = 'events_discarded'
def event_to_dict(event: BabeltraceEvent) -> DictEvent:
def event_to_dict(event: babeltrace.babeltrace.Event) -> DictEvent:
"""
Convert name, timestamp, and all other keys except those in IGNORED_FIELDS into a dictionary.