fixed test for finish method being put in the wrong place

1 files changed, 31 insertions, 1 deletions

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)
+      }
+    }
+  }
 }
+