Add Scalaland unit tests for Reg

1 files changed, 8 insertions, 1 deletions

diff --git a/src/test/scala/chiselTests/ChiselSpec.scala b/src/test/scala/chiselTests/ChiselSpec.scala
index 03f057d4..2d546d00 100644
--- a/src/test/scala/chiselTests/ChiselSpec.scala
+++ b/src/test/scala/chiselTests/ChiselSpec.scala
@@ -8,10 +8,17 @@ import org.scalacheck._
 import Chisel._
 import Chisel.testers._
 
-class ChiselPropSpec extends PropSpec with PropertyChecks {
+/** Common utility functions for Chisel unit tests. */
+trait ChiselRunners {
   def execute(t: => BasicTester): Boolean = TesterDriver.execute(t)
   def elaborate(t: => Module): Circuit = TesterDriver.elaborate(t)
+}
+
+/** Spec base class for BDD-style testers. */
+class ChiselFlatSpec extends FlatSpec with ChiselRunners with Matchers
 
+/** Spec base class for property-based testers. */
+class ChiselPropSpec extends PropSpec with ChiselRunners with PropertyChecks {
   def popCount(n: Long) = n.toBinaryString.count(_=='1')
 
   val smallPosInts = Gen.choose(1, 7)