more support for testing FPs

1 files changed, 8 insertions, 0 deletions

diff --git a/src/main/scala/Tester.scala b/src/main/scala/Tester.scala
index 91efc146..68fad15d 100644
--- a/src/main/scala/Tester.scala
+++ b/src/main/scala/Tester.scala
@@ -324,6 +324,14 @@ class ManualTester[+T <: Module]
   def expect (data: Bits, expected: Long): Boolean = {
     expect(data, BigInt(expected))
   }
+  def expect (data: Flo, expected: Double): Boolean = {
+    val got = peek(data)
+    expect(got == expected, "EXPECT " + data.debugName + " <- " + got + " == " + expected)
+  }
+  def expect (data: Dbl, expected: Double): Boolean = {
+    val got = peek(data)
+    expect(got == expected, "EXPECT " + data.debugName + " <- " + got + " == " + expected)
+  }
 
   /* Compare the floating point value of a node with an expected floating point value.
    * We will tolerate differences in the bottom bit.