Merge branch 'master' into lekcyjna/autumn-cleaning-2

kuznia-rdzeni · Oct 6, 2023 · 12fd8e4 · 12fd8e4
2 parents d2caf6a + a1556d5
commit 12fd8e4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/common.py b/test/common.py
@@ -157,7 +157,7 @@ def transform_methods_to_testbenchios(
 
         m.submodules.dut = self._dut
 
-        for name, attr in [(name, getattr(self._dut, name)) for name in dir(self._dut)]:
+        for name, attr in vars(self._dut).items():
             if guard_nested_collection(attr, Method) and attr:
                 tb_cont, mc = transform_methods_to_testbenchios(attr)
                 self._io[name] = tb_cont
@@ -166,7 +166,10 @@ def transform_methods_to_testbenchios(
         return m
 
     def debug_signals(self):
-        return [auto_debug_signals(io) for io in self._io.values()]
+        sigs = {"_dut": auto_debug_signals(self._dut)}
+        for name, io in self._io.items():
+            sigs[name] = auto_debug_signals(io)
+        return sigs
 
 
 class TestModule(Elaboratable):