From 2c70f05074305caef58004256443265892befed3 Mon Sep 17 00:00:00 2001
From: Daniel Benusovich <flyingmonkeys1996@gmail.com>
Date: Sat, 23 Feb 2019 13:53:11 -0800
Subject: [PATCH] Adding more logic to test

---
 TLB/test/test_cam.py | 51 ++++++++++++++++++++++++++++++++++++--------
 1 file changed, 42 insertions(+), 9 deletions(-)

diff --git a/TLB/test/test_cam.py b/TLB/test/test_cam.py
index 2e942c86..925a14eb 100644
--- a/TLB/test/test_cam.py
+++ b/TLB/test/test_cam.py
@@ -6,22 +6,34 @@ from nmigen.compat.sim import run_simulation
 
 from Cam import Cam
 
-from test_helper import check
+from test_helper import assert_eq, assert_ne
 
 def set_cam(dut, c, a, k, d):
+    print("asdf")
     yield dut.command.eq(c)
     yield dut.address.eq(a)
     yield dut.key_in.eq(k)
     yield dut.data_in.eq(d)
     yield
+    yield dut.command.eq(0)
+    yield dut.address.eq(0)
+    yield dut.key_in.eq(0)
+    yield dut.data_in.eq(0)
+    yield    
     
-def check_data_hit(dut, data_hit, op):
+def check_data_hit(dut, dh, op):
     out_dh = yield dut.data_hit
-    yield from check("Data Hit", out_dh, data_hit, op)
+    if op == 0:
+        assert_eq("Data Hit", out_dh, dh)
+    else:
+        assert_ne("Data Hit", out_dh, dh)
     
-def check_data(dut, data, op):
-    out_d = yield dut.data
-    yield from check("Data", out_d, data, op)  
+def check_data(dut, d, op):
+    out_d = yield dut.data_out
+    if op == 0:
+        assert_eq("Data", out_d, d)
+    else:
+        assert_ne("Data", out_d, d)  
     
 def check_all(dut, data_hit, data, dh_op, d_op):
     yield from check_data_hit(dut, data_hit, dh_op)
@@ -36,7 +48,7 @@ def testbench(dut):
     data = 0
     data_hit = 0
     yield from set_cam(dut, command, address, key, data)
-    yield from check_data_hit(dut, data_hit, 0)
+    #yield from check_data_hit(dut, data_hit, 0)
     
     # Search
     command = 3
@@ -45,16 +57,37 @@ def testbench(dut):
     data = 0
     data_hit = 0
     yield from set_cam(dut, command, address, key, data)
-    yield from check_data_hit(dut, data_hit, 0)    
+    #yield from check_data_hit(dut, data_hit, 0)    
     
     # Write Entry 0
+    command = 2
+    address = 0
+    key = 5
+    data = 4
+    data_hit = 0
+    yield from set_cam(dut, command, address, key, data)
+    #yield from check_data_hit(dut, data_hit, 0) 
+    
+    # Read Entry 0
     command = 1
     address = 0
+    key = 0
+    data = 4
+    data_hit = 0
+    yield from set_cam(dut, command, address, key, data)
+    #yield from check_all(dut, data_hit, data, 0, 0) 
+    
+    # Search 
+    command = 3
+    address = 0
     key = 5
     data = 4
     data_hit = 1
     yield from set_cam(dut, command, address, key, data)
-    yield from check_data_hit(dut, data_hit, 0)     
+    #yield from check_all(dut, data_hit, data, 0, 0)     
+    
+    yield 
+    
 
 if __name__ == "__main__":
     dut = Cam(4, 4, 4)
-- 
2.30.2