pull/35878/head
Shane Smiskol 5 days ago
parent 889e386dbc
commit ee6d54f38c
  1. 4
      selfdrive/test/process_replay/process_replay.py

@ -34,6 +34,7 @@ NUMPY_TOLERANCE = 1e-7
PROC_REPLAY_DIR = os.path.dirname(os.path.abspath(__file__)) PROC_REPLAY_DIR = os.path.dirname(os.path.abspath(__file__))
FAKEDATA = os.path.join(PROC_REPLAY_DIR, "fakedata/") FAKEDATA = os.path.join(PROC_REPLAY_DIR, "fakedata/")
class DummySocket: class DummySocket:
def __init__(self): def __init__(self):
self.data: list[bytes] = [] self.data: list[bytes] = []
@ -47,6 +48,7 @@ class DummySocket:
def send(self, data: bytes): def send(self, data: bytes):
self.data.append(data) self.data.append(data)
class LauncherWithCapture: class LauncherWithCapture:
def __init__(self, capture: ProcessOutputCapture, launcher: Callable): def __init__(self, capture: ProcessOutputCapture, launcher: Callable):
self.capture = capture self.capture = capture
@ -64,7 +66,7 @@ class ReplayContext:
self.main_pub = cfg.main_pub self.main_pub = cfg.main_pub
self.main_pub_drained = cfg.main_pub_drained self.main_pub_drained = cfg.main_pub_drained
self.unlocked_pubs = cfg.unlocked_pubs self.unlocked_pubs = cfg.unlocked_pubs
assert(len(self.pubs) != 0 or self.main_pub is not None) assert len(self.pubs) != 0 or self.main_pub is not None
def __enter__(self): def __enter__(self):
self.open_context() self.open_context()

Loading…
Cancel
Save