From 29dc51f8a1e1ebb3072fd6e7bd5fba7c331d029b Mon Sep 17 00:00:00 2001
From: chick
Date: Tue, 9 Feb 2016 11:32:19 -0800
Subject: fixed test for finish method being put in the wrong place

---
 src/test/scala/chiselTests/TesterDriverSpec.scala | 32 ++++++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

(limited to 'src/test/scala/chiselTests/TesterDriverSpec.scala')

diff --git a/src/test/scala/chiselTests/TesterDriverSpec.scala b/src/test/scala/chiselTests/TesterDriverSpec.scala
index ea1692f9..6271d382 100644
--- a/src/test/scala/chiselTests/TesterDriverSpec.scala
+++ b/src/test/scala/chiselTests/TesterDriverSpec.scala
@@ -5,6 +5,36 @@ package chiselTests
 /**
   * Created by chick on 2/9/16.
   */
-class TesterDriverSpec {
+import Chisel._
+import Chisel.testers.BasicTester
 
+/**
+  * Created by chick on 2/8/16.
+  */
+class FinishTester extends BasicTester {
+  var finish_was_run = false
+
+  override def finish(): Unit = {
+    finish_was_run = true
+  }
+}
+
+class TesterDriverSpec extends ChiselFlatSpec {
+  class DummyCircuit extends Module {
+    val io = new Bundle {
+      val in = Bool(INPUT)
+      val out = Bool(OUTPUT)
+    }
+  }
+
+  runTester {
+    new FinishTester {
+      val dut = new DummyCircuit
+
+      "Extending BasicTester" should "allow developer to have finish method run automatically" in {
+        assert(finish_was_run)
+      }
+    }
+  }
 }
+
-- 
cgit v1.2.3