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

diff --git a/migen/test/support.py b/migen/test/support.py
index 016373d7fab3d695ee0c566a8d7f59df9b46fa16..913462acdd677ae4e608cced2c4cbcee03d9ce5c 100644 (file)
--- a/migen/test/support.py
+++ b/migen/test/support.py
@@ -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)