Make submodule inputs void in ExpandWhens (#706)

1 files changed, 16 insertions, 0 deletions

diff --git a/src/test/scala/firrtlTests/CheckInitializationSpec.scala b/src/test/scala/firrtlTests/CheckInitializationSpec.scala
index cd6f464d..ef966ca0 100644
--- a/src/test/scala/firrtlTests/CheckInitializationSpec.scala
+++ b/src/test/scala/firrtlTests/CheckInitializationSpec.scala
@@ -55,4 +55,20 @@ class CheckInitializationSpec extends FirrtlFlatSpec {
       }
     }
   }
+  "Missing assignment to submodule port" should "trigger a PassException" in {
+    val input =
+      """circuit Test :
+        |  module Child :
+        |    input in : UInt<32>
+        |  module Test :
+        |    input p : UInt<1>
+        |    inst c of Child
+        |    when p :
+        |      c.in <= UInt(1)""".stripMargin
+    intercept[CheckInitialization.RefNotInitializedException] {
+      passes.foldLeft(parse(input)) {
+        (c: Circuit, p: Pass) => p.run(c)
+      }
+    }
+  }
 }