mirror of
https://github.com/m-labs/artiq.git
synced 2024-12-25 19:28:26 +08:00
rtio/sed: add output network simulation unittest
This commit is contained in:
parent
c5d6a2ba1a
commit
527b403bb1
56
artiq/gateware/test/rtio/test_sed_output_network.py
Normal file
56
artiq/gateware/test/rtio/test_sed_output_network.py
Normal file
@ -0,0 +1,56 @@
|
||||
import unittest
|
||||
|
||||
from migen import *
|
||||
|
||||
from artiq.gateware.rtio.sed import output_network
|
||||
|
||||
|
||||
LANE_COUNT = 8
|
||||
|
||||
|
||||
def simulate(input_events):
|
||||
dut = output_network.OutputNetwork(LANE_COUNT, LANE_COUNT*4, 3)
|
||||
output = []
|
||||
def gen():
|
||||
yield
|
||||
for n, input_event in enumerate(input_events):
|
||||
yield dut.input[n].valid.eq(1)
|
||||
yield dut.input[n].seqn.eq(n)
|
||||
for k, v in input_event.items():
|
||||
yield getattr(dut.input[n].payload, k).eq(v)
|
||||
yield
|
||||
for n in range(len(input_events)):
|
||||
yield dut.input[n].valid.eq(0)
|
||||
for i in range(6):
|
||||
yield
|
||||
for x in range(LANE_COUNT):
|
||||
if (yield dut.output[x].valid):
|
||||
d = {
|
||||
"replace_occured": (yield dut.output[x].replace_occured),
|
||||
"channel": (yield dut.output[x].payload.channel),
|
||||
"fine_ts": (yield dut.output[x].payload.fine_ts),
|
||||
"address": (yield dut.output[x].payload.address),
|
||||
"data": (yield dut.output[x].payload.data),
|
||||
}
|
||||
output.append(d)
|
||||
run_simulation(dut, gen())
|
||||
return output
|
||||
|
||||
|
||||
class TestOutputNetwork(unittest.TestCase):
|
||||
def test_replace(self):
|
||||
for n_events in range(2, LANE_COUNT+1):
|
||||
with self.subTest(n_events=n_events):
|
||||
input = [{"channel": 1, "address": i} for i in range(n_events)]
|
||||
output = simulate(input)
|
||||
expect = [{'replace_occured': 1, 'channel': 1, 'fine_ts': 0, 'address': n_events-1, 'data': 0}]
|
||||
self.assertEqual(output, expect)
|
||||
|
||||
def test_no_replace(self):
|
||||
for n_events in range(1, LANE_COUNT+1):
|
||||
with self.subTest(n_events=n_events):
|
||||
input = [{"channel": i, "address": i} for i in range(n_events)]
|
||||
output = simulate(input)
|
||||
expect = [{'replace_occured': 0, 'channel': i, 'fine_ts': 0, 'address': i, 'data': 0}
|
||||
for i in range(n_events)]
|
||||
self.assertEqual(output, expect)
|
Loading…
Reference in New Issue
Block a user