litex/migen/test/support.py

23 lines
528 B
Python

from migen.fhdl.std import *
from migen.sim.generic import run_simulation
from migen.fhdl import verilog
class SimBench(Module):
callback = None
def do_simulation(self, selfp):
if self.callback is not None:
return self.callback(self, selfp)
class SimCase:
TestBench = SimBench
def setUp(self, *args, **kwargs):
self.tb = self.TestBench(*args, **kwargs)
def test_to_verilog(self):
verilog.convert(self.tb)
def run_with(self, cb, ncycles=-1):
self.tb.callback = cb
run_simulation(self.tb, ncycles=ncycles)