summaryrefslogtreecommitdiff
path: root/tb/top/test_fifo.py
blob: 259d03df13bed96ddfb0bc8e93b0b9883486a8bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import itertools

import cocotb
from cocotb.clock import Clock
from cocotb.triggers import ClockCycles, RisingEdge, Timer
from cocotb_bus.drivers import BitDriver

@cocotb.test()
async def fifo(dut):
    await cocotb.start(Clock(dut.clk, 2).start())

    dut.in_valid.value = 0
    dut.out_ready.value = 0

    dut.rst_n.value = 1
    await Timer(1)
    dut.rst_n.value = 0
    await Timer(1)
    dut.rst_n.value = 1

    async def send():
        in_ = getattr(dut, 'in')
        in_ready = dut.in_ready
        in_valid = dut.in_valid

        val = 0
        while True:
            in_.value = val
            await RisingEdge(dut.clk)
            if in_valid.value and in_ready.value:
                val = (val + 1) & 0xff

    async def recv():
        out = dut.out
        out_ready = dut.out_ready
        out_valid = dut.out_valid

        val = 0
        while True:
            await RisingEdge(dut.clk)
            if out_valid.value and out_ready.value:
                assert out.value == val, f'expected {val}, got {out.value.integer}'
                val = (val + 1) & 0xff

    await cocotb.start(send())
    await cocotb.start(recv())

    await ClockCycles(dut.clk, 2)

    ready_driver = BitDriver(dut.out_ready, dut.clk)
    valid_driver = BitDriver(dut.in_valid, dut.clk)

    ready_driver.start((1, i % 5) for i in itertools.count())
    valid_driver.start((1 + (i % 2), (i + 1) % 3) for i in itertools.count())

    await ClockCycles(dut.clk, 1 << 16)