From 9e7ce6366454347e0ad912c7da0252070e5bb4a1 Mon Sep 17 00:00:00 2001
From: azidar
Date: Thu, 18 Aug 2016 17:43:05 -0700
Subject: Added test to check invalid bug was fixed

---
 src/test/scala/firrtlTests/ExpandWhensSpec.scala | 42 ++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

(limited to 'src/test')

diff --git a/src/test/scala/firrtlTests/ExpandWhensSpec.scala b/src/test/scala/firrtlTests/ExpandWhensSpec.scala
index 6219fd8c..82809dc8 100644
--- a/src/test/scala/firrtlTests/ExpandWhensSpec.scala
+++ b/src/test/scala/firrtlTests/ExpandWhensSpec.scala
@@ -32,9 +32,51 @@ import org.scalatest._
 import org.scalatest.prop._
 import firrtl._
 import firrtl.passes._
+import firrtl.ir._
+import firrtl.Parser.IgnoreInfo
 
 class ExpandWhensSpec extends FirrtlFlatSpec {
+  private def parse(input: String) = Parser.parse(input.split("\n").toIterator, IgnoreInfo)
+  private def executeTest(input: String, notExpected: String, passes: Seq[Pass]) = {
+    val c = passes.foldLeft(Parser.parse(input.split("\n").toIterator)) {
+      (c: Circuit, p: Pass) => p.run(c)
+    }
+    val lines = c.serialize.split("\n") map normalized
+
+    lines foreach { l =>
+      l.contains(notExpected) should be (false)
+    }
+  }
   "Expand Whens" should "compile and run" in {
     runFirrtlTest("ExpandWhens", "/passes/ExpandWhens")
   }
+  "Expand Whens" should "not emit INVALID" in {
+    val passes = Seq(
+      ToWorkingIR,
+      CheckHighForm,
+      ResolveKinds,
+      InferTypes,
+      CheckTypes,
+      Uniquify,
+      ResolveKinds,
+      InferTypes,
+      ResolveGenders,
+      CheckGenders,
+      InferWidths,
+      CheckWidths,
+      PullMuxes,
+      ExpandConnects,
+      RemoveAccesses,
+      ExpandWhens)
+    val input =
+  """|circuit Tester : 
+     |  module Tester :
+     |    input p : UInt<1>
+     |    when p :
+     |      wire a : {b : UInt<64>, c : UInt<64>}
+     |      a is invalid
+     |      a.b <= UInt<64>("h04000000000000000")""".stripMargin
+    val check = "INVALID"
+    executeTest(input, check, passes)
+  }
 }
-- 
cgit v1.2.3