X-Git-Url: https://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FOctet.hs;h=b3e6a0f5d3c1310f38443ca99dc49c43de516e5a;hb=a23ed449c784cdf81501c696bec10b1fbaeefc5f;hp=efc19d5607550e98da3bd47221fecb403e76d9e6;hpb=ea9bf8a98dbfc40ebad2af43b6594bac0a852dcd;p=hath.git

diff --git a/src/Octet.hs b/src/Octet.hs
index efc19d5..b3e6a0f 100644
--- 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 ]