Enable @chiselName on non-module classes (#1209)

1 files changed, 20 insertions, 1 deletions

diff --git a/src/test/scala/chiselTests/NamingAnnotationTest.scala b/src/test/scala/chiselTests/NamingAnnotationTest.scala
index 4576176a..e35c77c1 100644
--- a/src/test/scala/chiselTests/NamingAnnotationTest.scala
+++ b/src/test/scala/chiselTests/NamingAnnotationTest.scala
@@ -47,6 +47,21 @@ trait NamedModuleTester extends MultiIOModule {
     failures.toList
   }
 }
+@chiselName
+class OuterNamedNonModule {
+  val value = Wire(Bool())
+}
+
+@chiselName
+class NonModule {
+  val value = Wire(Bool())
+  @chiselName
+  class InnerNamedNonModule {
+    val value = Wire(Bool())
+  }
+  val inner = new InnerNamedNonModule
+  val outer = new OuterNamedNonModule
+}
 
 @chiselName
 class NamedModule extends NamedModuleTester {
@@ -84,10 +99,14 @@ class NamedModule extends NamedModuleTester {
     val noreturn = expectName(1.U + 2.U, "noreturn")
   }
 
-
   val test = expectName(FunctionMockup(), "test")
   val test2 = expectName(test +& 2.U, "test2")
   val test3 = expectName(ImplicitlyNamed(), "test3")
+  
+  val test4 = new NonModule
+  expectName(test4.value, "test4_value")
+  expectName(test4.inner.value, "test4_inner_value")
+  expectName(test4.outer.value, "test4_outer_value")
 
   // Test that contents of for loops are named
   for (i <- 0 until 1) {