Added Arbitrary instance definitions for the main data types.

[hath.git] / src / Octet.hs

diff --git a/src/Octet.hs b/src/Octet.hs
index efc19d5607550e98da3bd47221fecb403e76d9e6..b3e6a0f5d3c1310f38443ca99dc49c43de516e5a 100644 (file)
--- a/src/Octet.hs
+++ b/src/Octet.hs
@@ -1,6 +1,7 @@
 module Octet where
 
 import Test.HUnit
+import Test.QuickCheck
 
 import Bit
 
@@ -23,6 +24,21 @@ instance Show Octet where
     show oct = show (octet_to_int oct)
 
 
+instance Arbitrary Octet where
+    arbitrary = do
+      a1 <- arbitrary :: Gen Bit
+      a2 <- arbitrary :: Gen Bit
+      a3 <- arbitrary :: Gen Bit
+      a4 <- arbitrary :: Gen Bit
+      a5 <- arbitrary :: Gen Bit
+      a6 <- arbitrary :: Gen Bit
+      a7 <- arbitrary :: Gen Bit
+      a8 <- arbitrary :: Gen Bit
+      return (Octet a1 a2 a3 a4 a5 a6 a7 a8)
+
+    coarbitrary _ = variant 0
+
+
 -- Convert each bit to its integer value, and multiply by the
 -- appropriate power of two. Sum them up, and we should get an integer
 -- between 0 and 255.
@@ -73,10 +89,12 @@ max_octet = Octet One One One One One One One One
 
 
 -- HUnit Tests
+test_octet_from_int1 :: Test
 test_octet_from_int1 =
     TestCase $ assertEqual "octet_from_int 128 should parse as 10000000" oct1 (octet_from_int 128)
     where
       oct1 = Octet One Zero Zero Zero Zero Zero Zero Zero
 
 
+octet_tests :: [Test]
 octet_tests = [ test_octet_from_int1 ]