Make modular mult work

ChaCha20_Poly1305_64/sim/poly1305_friendly_modular_mult.py

@@ -8,6 +8,8 @@ from cocotb.queue import Queue
 
 from cocotbext.axi import AxiStreamBus, AxiStreamSource
 
+from modulo_theory import friendly_modular_mult
+
 import random
 
 PRIME = 2**130-5
@@ -48,7 +50,7 @@ class TB:
 
     async def write_input(self, data: int, h: int):
         await self.input_queue.put((data, h))
-        await self.expected_queue.put((data * h) % PRIME)
+        await self.expected_queue.put(friendly_modular_mult(h, data))
 
     async def run_input(self):
         while True:
@@ -76,7 +78,7 @@ async def test_sanity(dut):
 
     await tb.cycle_reset()
 
-    count = 1
+    count = 16
 
     for _ in range(count):
         await tb.write_input(random.randint(1,2**128-1), random.randint(0, 2**130-6))