migen/test/support: allow easy re-setUp of the testbench with different parameters

This commit is contained in:
Robert Jordens 2013-11-29 23:32:13 -07:00 committed by Sebastien Bourdeauducq
parent de830dc743
commit 4eef3b9399
1 changed files with 2 additions and 2 deletions

View File

@ -12,8 +12,8 @@ class SimBench(Module):
class SimCase(unittest.TestCase):
TestBench = SimBench
def setUp(self):
self.tb = self.TestBench()
def setUp(self, *args, **kwargs):
self.tb = self.TestBench(*args, **kwargs)
def test_to_verilog(self):
verilog.convert(self.tb)