Added Arbitrary instance definitions for the main data types.

[hath.git] / src / Bit.hs

diff --git a/src/Bit.hs b/src/Bit.hs
index d7fb679cfcaf079ebcf349b955d09a5b22da1f30..3799516c04c4514dd1a9241d0ea25ba753789438 100644 (file)
--- a/src/Bit.hs
+++ b/src/Bit.hs
@@ -1,5 +1,8 @@
 module Bit where
 
+import Test.QuickCheck
+
+
 data Bit = None | Zero | One
          deriving (Eq)
 
@@ -8,7 +11,12 @@ instance Show Bit where
     show Zero = "0"
     show One  = "1"
 
-                
+
+instance Arbitrary Bit where
+    arbitrary = elements [ Zero, One ]
+    coarbitrary _ = variant 0
+
+
 bit_to_int :: Bit -> Int
 bit_to_int None = -1
 bit_to_int Zero =  0