1 files changed, 43 insertions, 0 deletions
diff --git a/src/test/scala/ChiselTests/MemorySearch.scala b/src/test/scala/ChiselTests/MemorySearch.scala
new file mode 100644
index 00000000..e58005c2
--- /dev/null
+++ b/src/test/scala/ChiselTests/MemorySearch.scala
@@ -0,0 +1,43 @@
+package ChiselTests
+import Chisel._
+
+class MemorySearch extends Module {
+  val io = new Bundle {
+    val target  = UInt(INPUT,  4)
+    val en      = Bool(INPUT)
+    val done    = Bool(OUTPUT)
+    val address = UInt(OUTPUT, 3)
+  }
+  val vals  = Array(0, 4, 15, 14, 2, 5, 13)
+  val index = Reg(init = UInt(0, width = 3))
+  val elts  = Vec(vals.map(UInt(_,4)))
+  // val elts  = Mem(UInt(width = 32), 8)
+  val elt  = elts(index)
+  val end  = !io.en && ((elt === io.target) || (index === UInt(7)))
+  when (io.en) {
+    index := UInt(0)
+  } .elsewhen (!end) {
+    index := index +% UInt(1)
+  }
+  io.done    := end
+  io.address := index
+}
+
+class MemorySearchTester(c: MemorySearch) extends Tester(c) {
+  val list = c.vals
+  val n = 8
+  val maxT = n * (list.length + 3)
+  for (k <- 0 until n) {
+    val target = rnd.nextInt(16)
+    poke(c.io.en,     1)
+    poke(c.io.target, target)
+    step(1)
+    poke(c.io.en,     0)
+    do {
+      step(1)
+    } while (peek(c.io.done) == 0 && t < maxT)
+    val addr = peek(c.io.address).toInt
+    expect(addr == list.length || list(addr) == target, 
+           "LOOKING FOR " + target + " FOUND " + addr)
+  }
+}