Merge pull request #1 from ucb-bar/packagedir

Packagedir

1 files changed, 23 insertions, 0 deletions

diff --git a/src/test/scala/ChiselTests/Rom.scala b/src/test/scala/ChiselTests/Rom.scala
new file mode 100644
index 00000000..7c7eb1ac
--- /dev/null
+++ b/src/test/scala/ChiselTests/Rom.scala
@@ -0,0 +1,23 @@
+package ChiselTests
+import Chisel._
+
+class Rom extends Module {
+  val io = new Bundle {
+    val addr = UInt(INPUT,  4)
+    val out  = UInt(OUTPUT, 5)
+  }
+  val r = Vec(Range(0, 1 << 4).map(i => UInt(i * 2, width = 5)))
+  io.out := r(io.addr)
+}
+
+
+class RomTester(c: Rom) extends Tester(c) {
+  val r = Array.tabulate(1 << 4){ i => i * 2}
+  for (i <- 0 until 10) {
+    val a = rnd.nextInt(1 << 4)
+    poke(c.io.addr, a)
+    step(1)
+    expect(c.io.out, r(a))
+  }
+
+}