almost all tests work

[soc.git] / src / soc / TLB / test / test_set_associative_cache.py

diff --git a/src/soc/TLB/test/test_set_associative_cache.py b/src/soc/TLB/test/test_set_associative_cache.py
index 0641b556e9b77406fdaafa5d409e1950ca355b34..edec055b3545943f2c8c3e8b5399989af99544d6 100644 (file)
--- a/src/soc/TLB/test/test_set_associative_cache.py
+++ b/src/soc/TLB/test/test_set_associative_cache.py
@@ -1,8 +1,9 @@
 from nmigen.compat.sim import run_simulation
 
-from TLB.SetAssociativeCache import SetAssociativeCache
+from soc.TLB.SetAssociativeCache import SetAssociativeCache
+
+from soc.TestUtil.test_helper import assert_eq, assert_ne, assert_op
 
-from TestUtil.test_helper import assert_eq, assert_ne, assert_op
 
 def set_sac(dut, e, c, s, t, d):
     yield dut.enable.eq(e)
@@ -12,6 +13,7 @@ def set_sac(dut, e, c, s, t, d):
     yield dut.data_i.eq(d)
     yield
 
+
 def tbench(dut):
     enable = 1
     command = 2
@@ -29,10 +31,13 @@ def tbench(dut):
     yield from set_sac(dut, enable, command, cset, tag, data)
     yield
 
+
 def test_assoc_cache():
     dut = SetAssociativeCache(4, 4, 4, 4)
-    run_simulation(dut, tbench(dut), vcd_name="Waveforms/test_set_associative_cache.vcd")
+    run_simulation(dut, tbench(
+        dut), vcd_name="Waveforms/test_set_associative_cache.vcd")
     print("Set Associative Cache Unit Test Success")
 
+
 if __name__ == "__main__":
     test_assoc_cache()