Hardware
/
litedram
mirror of https://github.com/enjoy-digital/litedram.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test/test_adaption: use same DUT for up/down converter tests

This commit is contained in:
Florent Kermarrec 2019-07-13 10:52:41 +02:00
parent 9f9fed02f6
commit 8646b2e2c4
1 changed files with 22 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
test/test_adaptation.py
View File

@ -15,27 +15,28 @@ from test.common import *
from litex.gen.sim import * from litex.gen.sim import *
class TestAdaptation(unittest.TestCase): class ConverterDUT(Module):
def test_up_converter(self): def __init__(self, user_data_width, native_data_width):
class DUT(Module):
def __init__(self):
# write port and converter # write port and converter
self.write_user_port = LiteDRAMNativeWritePort(address_width=32, data_width=32) self.write_user_port = LiteDRAMNativeWritePort(address_width=32, data_width=user_data_width)
self.write_crossbar_port = LiteDRAMNativeWritePort(address_width=32, data_width=128) self.write_crossbar_port = LiteDRAMNativeWritePort(address_width=32, data_width=native_data_width)
write_converter = LiteDRAMNativePortConverter( write_converter = LiteDRAMNativePortConverter(
self.write_user_port, self.write_crossbar_port) self.write_user_port, self.write_crossbar_port)
self.submodules += write_converter self.submodules += write_converter
# read port and converter # read port and converter
self.read_user_port = LiteDRAMNativeReadPort(address_width=32, data_width=32) self.read_user_port = LiteDRAMNativeReadPort(address_width=32, data_width=user_data_width)
self.read_crossbar_port = LiteDRAMNativeReadPort(address_width=32, data_width=128) self.read_crossbar_port = LiteDRAMNativeReadPort(address_width=32, data_width=native_data_width)
read_converter = LiteDRAMNativePortConverter( read_converter = LiteDRAMNativePortConverter(
self.read_user_port, self.read_crossbar_port) self.read_user_port, self.read_crossbar_port)
self.submodules += read_converter self.submodules += read_converter
# memory # memory
self.memory = DRAMMemory(128, 128) self.memory = DRAMMemory(native_data_width, 128)
class TestAdaptation(unittest.TestCase):
def test_up_converter(self):
write_data = [seed_to_data(i, nbits=32) for i in range(16)] write_data = [seed_to_data(i, nbits=32) for i in range(16)]
read_data = [] read_data = []
@ -82,7 +83,7 @@ class TestAdaptation(unittest.TestCase):
for i in range(32): for i in range(32):
yield yield
dut = DUT() dut = ConverterDUT(user_data_width=32, native_data_width=128)
generators = [ generators = [
main_generator(dut.write_user_port, dut.read_user_port), main_generator(dut.write_user_port, dut.read_user_port),
read_handler(dut.read_user_port), read_handler(dut.read_user_port),
@ -93,25 +94,6 @@ class TestAdaptation(unittest.TestCase):
self.assertEqual(write_data, read_data) self.assertEqual(write_data, read_data)
def test_down_converter(self): def test_down_converter(self):
class DUT(Module):
def __init__(self):
# write port and converter
self.write_user_port = LiteDRAMNativeWritePort(address_width=32, data_width=64)
self.write_crossbar_port = LiteDRAMNativeWritePort(address_width=32, data_width=32)
write_converter = LiteDRAMNativePortConverter(
self.write_user_port, self.write_crossbar_port)
self.submodules += write_converter
# read port and converter
self.read_user_port = LiteDRAMNativeReadPort(address_width=32, data_width=64)
self.read_crossbar_port = LiteDRAMNativeReadPort(address_width=32, data_width=32)
read_converter = LiteDRAMNativePortConverter(
self.read_user_port, self.read_crossbar_port)
self.submodules += read_converter
# memory
self.memory = DRAMMemory(32, 128)
write_data = [seed_to_data(i, nbits=64) for i in range(8)] write_data = [seed_to_data(i, nbits=64) for i in range(8)]
read_data = [] read_data = []
@ -154,7 +136,7 @@ class TestAdaptation(unittest.TestCase):
for i in range(32): for i in range(32):
yield yield
dut = DUT() dut = ConverterDUT(user_data_width=64, native_data_width=32)
generators = [ generators = [
main_generator(dut.write_user_port, dut.read_user_port), main_generator(dut.write_user_port, dut.read_user_port),
read_handler(dut.read_user_port), read_handler(dut.read_user_port),