1 files changed, 53 insertions, 0 deletions
diff --git a/hdl/testing/memtest.py b/hdl/testing/memtest.py
new file mode 100644
index 0000000..459450a
--- /dev/null
+++ b/hdl/testing/memtest.py
@@ -0,0 +1,53 @@
+from amaranth import *
+from amaranth.cli import main
+
+from amaranth.back import verilog, cxxrtl, rtlil
+
+from hdl.config import *
+from hdl.utils import *
+
+class RegisterFile(Elaboratable):
+    def __init__(self, sim=True):
+        self.adr   = Signal(12)
+        self.dat_r = Signal(16)
+        self.dat_w = Signal(16)
+        self.we    = Signal()
+        self.mem   = Memory(width=16, depth=2**len(self.adr), no_init=True, simulate=sim)
+
+        self.ports = {'in': [self.adr, self.we], 'out': [self.dat_r]}
+
+    def elaborate(self, platform):
+        m = Module()
+        m.submodules.rdport = rdport = self.mem.read_port()
+        m.submodules.wrport = wrport = self.mem.write_port()
+        m.d.comb += [
+            rdport.addr.eq(self.adr),
+            self.dat_r.eq(rdport.data),
+            wrport.addr.eq(self.adr),
+            wrport.data.eq(self.dat_w),
+            wrport.en.eq(self.we),
+        ]
+        return m
+
+
+def test_mem():
+    dut = RegisterFile(sim=True)
+    def proc():
+        yield from eval()
+        yield dut.adr.eq(0)
+        yield dut.dat_w.eq(27)
+        yield dut.we.eq(1)
+        yield from step()
+        yield dut.we.eq(0)
+        yield from step()
+        data_out = yield dut.dat_r
+        print(f'Data out: {data_out}')
+    sim(dut, proc)
+
+
+if __name__ == "__main__":
+    hdl = RegisterFile()
+    # test_mem()
+    out = verilog.convert(hdl, ports=hdl.ports['in'] + hdl.ports['out'])
+    with open(os.path.join(VERILOG_DIR, os.path.basename("memtest.py".replace('.py', '.v'))), 'w') as f:
+        f.write(out)
+\ No newline at end of file