test_runner.py: add needed imports

diff --git a/src/soc/simple/test/test_runner.py b/src/soc/simple/test/test_runner.py
index 381395fad2c6830cd57ec20cf462ba3eaad0400d..8700b18a90e6e1dc1c7fac8638e09827d88c205a 100644 (file)
--- a/src/soc/simple/test/test_runner.py
+++ b/src/soc/simple/test/test_runner.py
@@ -30,6 +30,8 @@ from soc.simple.test.test_core import (setup_regs, check_regs,
 from soc.fu.compunits.test.test_compunit import (setup_test_memory,
                                                  check_sim_memory)
 from soc.debug.dmi import DBGCore, DBGCtrl, DBGStat
+from nmutil.util import wrap
+from soc.experiment.test.test_mmu_dcache import wb_get
 
 
 def setup_i_memory(imem, startaddr, instructions):