Update for QuickCheck 2.

[hath.git] / src / Bit.hs

diff --git a/src/Bit.hs b/src/Bit.hs
index d7fb679cfcaf079ebcf349b955d09a5b22da1f30..a4241bc2a72b8a062649def5d1daa40f0a13f030 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,11 @@ instance Show Bit where
     show Zero = "0"
     show One  = "1"
 
-                
+
+instance Arbitrary Bit where
+    arbitrary = elements [ Zero, One ]
+
+
 bit_to_int :: Bit -> Int
 bit_to_int None = -1
 bit_to_int Zero =  0