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