use yield from on StateRunners

diff --git a/src/soc/simple/test/test_runner.py b/src/soc/simple/test/test_runner.py
index 0eb330783320fd8bd5493a05593f8c2a98d8f150..ee880352ffbf1567890e68036a334b56f0eba31c 100644 (file)
--- a/src/soc/simple/test/test_runner.py
+++ b/src/soc/simple/test/test_runner.py
@@ -394,7 +394,7 @@ class TestRunner(FHDLTestCase):
             # there is a fallback which can be "yielded".
 
             if self.run_sim:
-                simrun.setup_during_test() # TODO, some arguments?
+                yield from simrun.setup_during_test() # TODO, some arguments?
 
             if self.run_hdl:
                 yield from hdlrun.setup_during_test()
@@ -410,7 +410,7 @@ class TestRunner(FHDLTestCase):
                     # TODO https://bugs.libre-soc.org/show_bug.cgi?id=686#c73
 
                     if self.run_sim:
-                        simrun.prepare_for_test(test)
+                        yield from simrun.prepare_for_test(test)
 
                     if self.run_hdl:
                         yield from hdlrun.prepare_for_test(test)
@@ -501,20 +501,20 @@ class TestRunner(FHDLTestCase):
                 # TODO https://bugs.libre-soc.org/show_bug.cgi?id=686#c73
 
                 if self.run_sim:
-                    simrun.end_test() # TODO, some arguments?
+                    yield from simrun.end_test() # TODO, some arguments?
 
                 if self.run_hdl:
                     yield from hdlrun.end_test()
 
-        ###### END OF EVERYTHING (but none needs doing, still call fn) #######
-        # StateRunner.cleanup()
-        # TODO https://bugs.libre-soc.org/show_bug.cgi?id=686#c73
+            ###### END OF EVERYTHING (but none needs doing, still call fn) ####
+            # StateRunner.cleanup()
+            # TODO https://bugs.libre-soc.org/show_bug.cgi?id=686#c73
 
-        if self.run_sim:
-            simrun.cleanup() # TODO, some arguments?
+            if self.run_sim:
+                yield from simrun.cleanup() # TODO, some arguments?
 
-        if self.run_hdl:
-            hdlrun.cleanup()
+            if self.run_hdl:
+                yield from hdlrun.cleanup()
 
         styles = {
             'dec': {'base': 'dec'},