Whatever I was working on

ChaCha20_Poly1305_64/sim/poly1305_stage.py

@@ -5,6 +5,7 @@ import cocotb
 from cocotb.clock import Clock
 from cocotb.triggers import Timer, RisingEdge, FallingEdge
 from cocotb.queue import Queue
+from cocotb.handle import Immediate
 
 from cocotbext.axi import AxiStreamBus, AxiStreamSource
 
@@ -29,7 +30,7 @@ class TB:
         self.expected_queue = Queue()
         self.output_queue = Queue()
 
-        cocotb.start_soon(Clock(self.dut.i_clk, CLK_PERIOD, units="ns").start())
+        cocotb.start_soon(Clock(self.dut.i_clk, CLK_PERIOD, unit="ns").start())
 
         cocotb.start_soon(self.run_input())
         cocotb.start_soon(self.run_output())
@@ -41,7 +42,7 @@ class TB:
         await self._cycle_reset(self.dut.i_rst, self.dut.i_clk)
 
     async def _cycle_reset(self, rst, clk):
-        rst.setimmediatevalue(0)
+        rst.value = Immediate(0)
         await RisingEdge(clk)
         await RisingEdge(clk)
         rst.value = 1
@@ -82,7 +83,7 @@ class TB:
         while True:
             await RisingEdge(self.dut.i_clk)
             if self.dut.o_valid.value:
-                await self.output_queue.put(self.dut.o_result.value.integer)
+                await self.output_queue.put(self.dut.o_result.value.to_unsigned())
 
 @cocotb.test
 async def test_sanity(dut):