From c5187a33bdba8876c2210c284a5774444e1dcc8c Mon Sep 17 00:00:00 2001
From: Luke Kenneth Casson Leighton <lkcl@lkcl.net>
Date: Sat, 25 Sep 2021 14:02:15 +0100
Subject: [PATCH] add dummy call to simrun and end_test()

---
 src/soc/simple/test/test_runner.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/soc/simple/test/test_runner.py b/src/soc/simple/test/test_runner.py
index 31c6b7b4..c8c3e827 100644
--- a/src/soc/simple/test/test_runner.py
+++ b/src/soc/simple/test/test_runner.py
@@ -460,6 +460,9 @@ class TestRunner(FHDLTestCase):
                 ###### END OF A TEST #######
                 # StateRunner.end_test()
 
+                if self.run_sim:
+                    simrun.end_test() # TODO, some arguments?
+
                 if self.run_hdl:
                     # stop at end
                     yield from set_dmi(hdlrun.dmi, DBGCore.CTRL, 1<<DBGCtrl.STOP)
@@ -495,6 +498,10 @@ class TestRunner(FHDLTestCase):
         ###### END OF EVERYTHING (but none needs doing, still call fn) #######
         # StateRunner.cleanup()
 
+        if self.run_sim:
+            simrun.cleanup() # TODO, some arguments?
+
+
         styles = {
             'dec': {'base': 'dec'},
             'bin': {'base': 'bin'},
-- 
2.30.2