From fe90b176fc9f72d112d96662ab0847811335a794 Mon Sep 17 00:00:00 2001 From: mferris Date: Tue, 29 Jul 2025 13:16:30 -0300 Subject: [PATCH] add option to force waiting into send_message --- src/caveat/caveatbench.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/caveat/caveatbench.py b/src/caveat/caveatbench.py index 6cbc89e..9141640 100644 --- a/src/caveat/caveatbench.py +++ b/src/caveat/caveatbench.py @@ -154,10 +154,14 @@ async def add_receiver_axis(self, label: str, clk, prefix: str='', if monitor: self.monitors[label] = CaveatAxiStreamMonitor(bus, clk) - async def send_message(self, sender_name, message): + async def send_message(self, sender_name, message, force_wait=False): """Send an integer, a list of integers, a byte, or a bytearray to DUT. """ - self.sources[sender_name].send_nowait(list(message)) + if force_wait: + await self.sources[sender_name].send(list(message)) + await self.sources[sender_name].wait() + else: + self.sources[sender_name].send_nowait(list(message)) async def read_message(self, receiver_name): """Read value out from specified receiver, returns list of integers