forked from M-Labs/artiq
1
0
Fork 0

test: change base address in DMA simulation testbench

This commit is contained in:
Sebastien Bourdeauducq 2017-03-31 13:17:00 +08:00
parent 245e186347
commit 200c499114
1 changed files with 39 additions and 15 deletions

View File

@ -23,7 +23,7 @@ def encode_record(channel, timestamp, address, data):
r += encode_n(channel, 3, 3)
r += encode_n(timestamp, 8, 8)
r += encode_n(address, 2, 2)
r += encode_n(data, 4, 64)
r += encode_n(data, 1, 64)
return encode_n(len(r)+1, 1, 1) + r
@ -37,24 +37,45 @@ def pack(x, size):
n |= x[j]
except IndexError:
pass
# print("{:0128x}".format(n))
r.append(n)
return r
test_writes = [
def encode_sequence(writes, ws):
sequence = [b for write in writes for b in encode_record(*write)]
sequence.append(0)
return pack(sequence, ws)
test_writes1 = [
(0x01, 0x23, 0x12, 0x33),
(0x901, 0x902, 0x911, 0xeeeeeeeeeeeeeefffffffffffffffffffffffffffffff28888177772736646717738388488),
(0x81, 0x288, 0x88, 0x8888)
]
test_writes2 = [
(0x10, 0x10000, 0x20, 0x77),
(0x11, 0x10001, 0x22, 0x7777),
(0x12, 0x10002, 0x30, 0x777777),
(0x13, 0x10003, 0x40, 0x77777788),
(0x14, 0x10004, 0x50, 0x7777778899),
]
prng = random.Random(0)
class TB(Module):
def __init__(self, ws):
sequence = [b for write in test_writes for b in encode_record(*write)]
sequence.append(0)
# print(sequence)
sequence = pack(sequence, ws)
sequence1 = encode_sequence(test_writes1, ws)
sequence2 = encode_sequence(test_writes2, ws)
offset = 512//ws
assert len(sequence1) < offset
sequence = (
sequence1 +
[prng.randrange(2**(ws*8)) for _ in range(offset-len(sequence1))] +
sequence2)
bus = wishbone.Interface(ws*8)
self.submodules.memory = wishbone.SRAM(
@ -64,16 +85,19 @@ class TB(Module):
class TestDMA(unittest.TestCase):
def test_dma_noerror(self):
prng = random.Random(0)
ws = 64
tb = TB(ws)
def do_dma():
for i in range(2):
yield from tb.dut.enable.write(1)
def do_dma(address):
yield from tb.dut.dma.base_address.write(address)
yield from tb.dut.enable.write(1)
yield
while ((yield from tb.dut.enable.read())):
yield
while ((yield from tb.dut.enable.read())):
yield
def do_writes():
yield from do_dma(0)
yield from do_dma(512)
received = []
@passive
@ -98,5 +122,5 @@ class TestDMA(unittest.TestCase):
self.fail("unexpected RTIO command")
yield
run_simulation(tb, [do_dma(), rtio_sim()])
self.assertEqual(received, test_writes*2)
run_simulation(tb, [do_writes(), rtio_sim()])
self.assertEqual(received, test_writes1 + test_writes2)