From 2b71a50abcf71ff2f90802ba1b06d94428e550f1 Mon Sep 17 00:00:00 2001
From: Sequencer
Date: Fri, 10 Apr 2020 14:51:41 +0000
Subject: Expose checkTypeEquivalence as a public API (#1402)

---
 src/test/scala/chiselTests/RecordSpec.scala | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src/test/scala')

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 }
+  }
 }
-- 
cgit v1.2.3