Expose checkTypeEquivalence as a public API (#1402)

1 files changed, 12 insertions, 0 deletions

diff --git a/src/test/scala/chiselTests/RecordSpec.scala b/src/test/scala/chiselTests/RecordSpec.scala
index bf6b92eb..41242981 100644
--- a/src/test/scala/chiselTests/RecordSpec.scala
+++ b/src/test/scala/chiselTests/RecordSpec.scala
@@ -99,6 +99,14 @@ trait RecordSpecUtils {
     assert(wire("0").asUInt === 123.U)
     stop()
   }
+
+  class RecordTypeTester extends BasicTester {
+    val wire0 = Wire(new CustomBundle("0"-> UInt(32.W)))
+    val wire1 = Reg(new CustomBundle("0"-> UInt(32.W)))
+    val wire2 = Wire(new CustomBundle("1"-> UInt(32.W)))
+    require(DataMirror.checkTypeEquivalence(wire0, wire1))
+    require(!DataMirror.checkTypeEquivalence(wire1, wire2))
+  }
 }
 
 class RecordSpec extends ChiselFlatSpec with RecordSpecUtils {
@@ -146,4 +154,8 @@ class RecordSpec extends ChiselFlatSpec with RecordSpecUtils {
       io := wire
     })
   }
+
+  "CustomBundle" should "check the types" in {
+    elaborate { new RecordTypeTester }
+  }
 }