@@ -1,32 +1,34 @@
1−{-# LANGUAGE CPP, FlexibleInstances, FlexibleContexts, TupleSections #-}
1+{-# LANGUAGE FlexibleInstances, FlexibleContexts, TupleSections #-}
3−module Main ( main ) where
3+module Main where
5−import Data.Binary ( Binary(..) )
5+import Data.Binary ( encode, Binary(..) )
6−import Data.Binary.Get ( runGet, runGetIncremental, pushChunks, Decoder(..) )
6+import Data.Binary.Get ( runGet )
7import Data.Binary.Put ( runPut )
7import Data.Binary.Put ( runPut )
9+import qualified Data.Binary.Get as BG ( getWord8, getWord16be, getWord32be, getWord64be )
10+import qualified Data.Binary.Put as BP ( putWord8, putWord16be, putWord32be, putWord64be )
9import Data.Binary.Bits
12import Data.Binary.Bits
10import Data.Binary.Bits.Get
13import Data.Binary.Bits.Get
11import Data.Binary.Bits.Put
14import Data.Binary.Bits.Put
12−import Data.Binary.Bits.BitOrder
14import qualified Data.ByteString as B
16import qualified Data.ByteString as B
15−import qualified Data.ByteString.Lazy as L
17import Control.Applicative
18import Control.Applicative
18import Data.Bits
19import Data.Bits
19import Data.Word
21import Data.Word
20import Foreign.Storable
22import Foreign.Storable
21−import Data.Traversable (traverse)
23+import System.Random
22−import Data.Foldable (traverse_)
25+import Test.Framework.Options ( TestOptions'(..) )
24import Test.Framework.Providers.QuickCheck2 ( testProperty )
26import Test.Framework.Providers.QuickCheck2 ( testProperty )
25import Test.Framework.Runners.Console ( defaultMain )
27import Test.Framework.Runners.Console ( defaultMain )
28+import Test.Framework.Runners.Options ( RunnerOptions'(..) )
26import Test.Framework ( Test, testGroup )
29import Test.Framework ( Test, testGroup )
27import Test.QuickCheck
30import Test.QuickCheck
30main = defaultMain tests
32main = defaultMain tests
32tests :: [Test]
34tests :: [Test]
@@ -37,29 +39,6 @@ tests =
37 , testGroup "Custom test cases"
39 , testGroup "Custom test cases"
38 [ testProperty "prop_composite_case" prop_composite_case ]
40 [ testProperty "prop_composite_case" prop_composite_case ]
40− , testGroup "getByteString"
41− [ testProperty "prop_getByteString_negative" prop_getByteString_negative
42− , testProperty "prop_putByteString_getByteString" (prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property)
43− , testProperty "prop_putByteString_getByteString_many" (prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property)
46− , testGroup "getLazyByteString"
47− [ testProperty "getLazyByteString == getByteString"
48− prop_getLazyByteString_equal_to_ByteString
49− , testProperty "getLazyByteString == getByteString (with shift)"
50− prop_getLazyByteString_equal_to_ByteString2
53− , testGroup "isEmpty"
54− [ testProperty "prop_isEmptyOfEmptyEmpty" prop_isEmptyOfEmptyEmpty
55− , testProperty "prop_isEmptyOfNonEmptyEmpty" prop_isEmptyOfNonEmptyEmpty
56− , testProperty "prop_isEmptyOfConsumedEmpty" prop_isEmptyOfConsumedEmpty
57− , testProperty "prop_isEmptyOfNotConsumedNotEmpty" prop_isEmptyOfNotConsumedNotEmpty
61− [ testProperty "monadic fail" prop_fail ]
63 , testGroup "prop_bitput_with_get_from_binary"
42 , testGroup "prop_bitput_with_get_from_binary"
64 [ testProperty "Word8" (prop_bitput_with_get_from_binary :: W [Word8] -> Property)
43 [ testProperty "Word8" (prop_bitput_with_get_from_binary :: W [Word8] -> Property)
65 , testProperty "Word16" (prop_bitput_with_get_from_binary :: W [Word16] -> Property)
44 , testProperty "Word16" (prop_bitput_with_get_from_binary :: W [Word16] -> Property)
@@ -89,10 +68,10 @@ tests =
91 , testGroup "prop_put_with_bitreq"
70 , testGroup "prop_put_with_bitreq"
92− [ testProperty "Word8" (prop_putget_with_bitreq :: BitOrder -> W Word8 -> Property)
71+ [ testProperty "Word8" (prop_putget_with_bitreq :: W Word8 -> Property)
93− , testProperty "Word16" (prop_putget_with_bitreq :: BitOrder -> W Word16 -> Property)
72+ , testProperty "Word16" (prop_putget_with_bitreq :: W Word16 -> Property)
94− , testProperty "Word32" (prop_putget_with_bitreq :: BitOrder -> W Word32 -> Property)
73+ , testProperty "Word32" (prop_putget_with_bitreq :: W Word32 -> Property)
95− , testProperty "Word64" (prop_putget_with_bitreq :: BitOrder -> W Word64 -> Property)
74+ , testProperty "Word64" (prop_putget_with_bitreq :: W Word64 -> Property)
98 , testGroup "prop_putget_list_simple"
77 , testGroup "prop_putget_list_simple"
@@ -103,13 +82,6 @@ tests =
103 , testProperty "Word64" (prop_putget_list_simple :: W [Word64] -> Property)
82 , testProperty "Word64" (prop_putget_list_simple :: W [Word64] -> Property)
106− , testGroup "prop_putget_applicative_with_bitreq"
107− [ testProperty "Word8" (prop_putget_applicative_with_bitreq :: W [(Word8,Word8,Word8)] -> Property)
108− , testProperty "Word16" (prop_putget_applicative_with_bitreq :: W [(Word16,Word16,Word16)] -> Property)
109− , testProperty "Word32" (prop_putget_applicative_with_bitreq :: W [(Word32,Word32,Word32)] -> Property)
110− , testProperty "Word64" (prop_putget_applicative_with_bitreq :: W [(Word64,Word64,Word64)] -> Property)
113 , testGroup "prop_putget_list_with_bitreq"
85 , testGroup "prop_putget_list_with_bitreq"
114 [ testProperty "Word8" (prop_putget_list_with_bitreq :: W [Word8] -> Property)
86 [ testProperty "Word8" (prop_putget_list_with_bitreq :: W [Word8] -> Property)
115 , testProperty "Word16" (prop_putget_list_with_bitreq :: W [Word16] -> Property)
87 , testProperty "Word16" (prop_putget_list_with_bitreq :: W [Word16] -> Property)
@@ -117,74 +89,27 @@ tests =
117 , testProperty "Word64" (prop_putget_list_with_bitreq :: W [Word64] -> Property)
89 , testProperty "Word64" (prop_putget_list_with_bitreq :: W [Word64] -> Property)
119 , testGroup "prop_bitget_bytestring_interspersed"
91 , testGroup "prop_bitget_bytestring_interspersed"
120− [ testProperty "Word8" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word8 -> [B.ByteString] -> Property)
92+ [ testProperty "Word8" (prop_bitget_bytestring_interspersed :: W Word8 -> [B.ByteString] -> Property)
121− , testProperty "Word16" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word16 -> [B.ByteString] -> Property)
93+ , testProperty "Word16" (prop_bitget_bytestring_interspersed :: W Word16 -> [B.ByteString] -> Property)
122− , testProperty "Word32" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word32 -> [B.ByteString] -> Property)
94+ , testProperty "Word32" (prop_bitget_bytestring_interspersed :: W Word32 -> [B.ByteString] -> Property)
123− , testProperty "Word64" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word64 -> [B.ByteString] -> Property)
95+ , testProperty "Word64" (prop_bitget_bytestring_interspersed :: W Word64 -> [B.ByteString] -> Property)
125− , testGroup "Simulate programs"
126− [ testProperty "primitive" prop_primitive
127− , testProperty "many primitives in sequence" prop_program
131−prop_isEmptyOfEmptyEmpty :: Bool
99+{-
132−prop_isEmptyOfEmptyEmpty = runGet (runBitGet isEmpty) L.empty
100+ -- these tests use the R structure
134−prop_isEmptyOfNonEmptyEmpty :: L.ByteString -> Property
102+ -- quickCheck prop_Word32_from_2_Word16
135−prop_isEmptyOfNonEmptyEmpty bs =
103+ -- quickCheck prop_Word32_from_Word8_and_Word16
136− not (L.null bs) ==> not (runGet (runBitGet isEmpty) bs)
104+-}
138−prop_isEmptyOfConsumedEmpty :: L.ByteString -> Property
139−prop_isEmptyOfConsumedEmpty bs =
141− runGet (runBitGet (getByteString n >> isEmpty)) bs
142− where n = fromIntegral $ L.length bs
144−prop_isEmptyOfNotConsumedNotEmpty :: L.ByteString -> Int -> Property
145−prop_isEmptyOfNotConsumedNotEmpty bs n =
146− (fromIntegral n) < L.length bs && not (L.null bs) ==>
147− not (runGet (runBitGet (getByteString n >> isEmpty)) bs)
149−prop_getLazyByteString_equal_to_ByteString :: L.ByteString -> Int -> Property
150−prop_getLazyByteString_equal_to_ByteString bs n =
151− (fromIntegral n) <= L.length bs ==>
152− runGet (runBitGet (getLazyByteString (fromIntegral n))) bs ==
153− (L.fromChunks . (:[]) $ runGet (runBitGet (getByteString n)) bs)
155−prop_getLazyByteString_equal_to_ByteString2 :: L.ByteString -> Int -> Property
156−prop_getLazyByteString_equal_to_ByteString2 bs n =
157− (L.length bs > 1) && (fromIntegral n) < L.length bs ==>
158− runGet (runBitGet (getWord8 2 >> getLazyByteString (fromIntegral n))) bs ==
159− (L.fromChunks . (:[]) $ runGet (runBitGet (getWord8 2 >> getByteString n)) bs)
161−prop_getByteString_negative :: Int -> Property
162−prop_getByteString_negative n =
164− runGet (runBitGet (getByteString n)) L.empty == B.empty
166−prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property
167−prop_putByteString_getByteString bo bs = property $ bs' == bs
170− w = runPut (runBitPut (withBitOrder bo (putByteString bs)))
171− bs' = runGet (runBitGet (withBitOrder bo (getByteString n))) w
173−prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property
174−prop_putByteString_getByteString_many bo bs = property $ bs' == bs
176− n = fmap B.length bs
177− w = runPut (runBitPut (withBitOrder bo (traverse_ putByteString bs)))
178− bs' = runGet (runBitGet (withBitOrder bo (traverse getByteString n))) w
181−prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => BitOrder -> W a -> Property
106+prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => W a -> Property
182−prop_putget_with_bitreq bo (W w) = property $
107+prop_putget_with_bitreq (W w) = property $
183 -- write all words with as many bits as it's required
108 -- write all words with as many bits as it's required
184 let p = putBits (bitreq w) w
109 let p = putBits (bitreq w) w
185 g = getBits (bitreq w)
110 g = getBits (bitreq w)
186− lbs = runPut (runBitPut (withBitOrder bo p))
111+ lbs = runPut (runBitPut p)
187− w' = runGet (runBitGet (withBitOrder bo g)) lbs
112+ w' = runGet (runBitGet g) lbs
188 in w == w'
113 in w == w'
190-- | Write a list of items. Each item is written with the maximum amount of
115-- | Write a list of items. Each item is written with the maximum amount of
@@ -192,7 +117,7 @@ prop_putget_with_bitreq bo (W w) = property $
192prop_putget_list_simple :: (BinaryBit a, Eq a, Storable a) => W [a] -> Property
117prop_putget_list_simple :: (BinaryBit a, Eq a, Storable a) => W [a] -> Property
193prop_putget_list_simple (W ws) = property $
118prop_putget_list_simple (W ws) = property $
194 let s = sizeOf (head ws) * 8
119 let s = sizeOf (head ws) * 8
195− p = mapM_ (putBits s) ws
120+ p = mapM_ (\v -> putBits s v) ws
196 g = mapM (const (getBits s)) ws
121 g = mapM (const (getBits s)) ws
197 lbs = runPut (runBitPut p)
122 lbs = runPut (runBitPut p)
198 ws' = runGet (runBitGet g) lbs
123 ws' = runGet (runBitGet g) lbs
@@ -211,18 +136,6 @@ prop_putget_list_with_bitreq (W ws) = property $
212 bitlist = map bitreq ws
137 bitlist = map bitreq ws
214−prop_putget_applicative_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => W [(a,a,a)] -> Property
215−prop_putget_applicative_with_bitreq (W ts) = property $
216− let p = mapM_ (\(a,b,c) -> do putBits (bitreq a) a
217− putBits (bitreq b) b
218− putBits (bitreq c) c) ts
219− g = mapM (\(a,b,c) -> (,,) <$> getBits a <*> getBits b <*> getBits c) bitlist
220− lbs = runPut (runBitPut p)
221− ts' = runGet (runBitGet g) lbs
224− bitlist = map (\(a,b,c) -> (bitreq a, bitreq b, bitreq c)) ts
226-- | Write bits using this library, and read them back using the binary
139-- | Write bits using this library, and read them back using the binary
227-- library.
140-- library.
228prop_bitput_with_get_from_binary :: (BinaryBit a, Binary a, Storable a, Eq a) => W [a] -> Property
141prop_bitput_with_get_from_binary :: (BinaryBit a, Binary a, Storable a, Eq a) => W [a] -> Property
@@ -246,52 +159,33 @@ prop_bitget_with_put_from_binary (W ws) = property $
246 in ws == ws'
159 in ws == ws'
248-- | Write each 'ByteString' with a variable sized value as a separator.
161-- | Write each 'ByteString' with a variable sized value as a separator.
249−prop_bitget_bytestring_interspersed :: (BinaryBit a, Binary a, Num a, Ord a, Bits a) => BitOrder -> W a -> [B.ByteString] -> Property
162+prop_bitget_bytestring_interspersed :: (BinaryBit a, Binary a, Num a, Ord a, Bits a) => W a -> [B.ByteString] -> Property
250−prop_bitget_bytestring_interspersed bo (W ws) bss = property $
163+prop_bitget_bytestring_interspersed (W ws) bss = property $
251 let p = mapM_ (\bs -> putBits (bitreq ws) ws >> putByteString bs) bss
164 let p = mapM_ (\bs -> putBits (bitreq ws) ws >> putByteString bs) bss
252− g = mapM (\bs -> (,) <$> getBits (bitreq ws) <*> getByteString (B.length bs)) bss
165+ g = mapM (\bs -> (,) <$> (getBits (bitreq ws)) <*> (getByteString (B.length bs))) bss
253− lbs = runPut (runBitPut (withBitOrder bo p))
166+ lbs = runPut (runBitPut p)
254− r = runGet (runBitGet (withBitOrder bo g)) lbs
167+ r = runGet (runBitGet g) lbs
255 in map (ws,) bss == r
168 in map (ws,) bss == r
258−prop_fail :: L.ByteString -> String -> Property
259−prop_fail lbs errMsg0 = forAll (choose (0, 8 * L.length lbs)) $ \len ->
260− let (bytes,bits) = len `divMod` 8
261− expectedBytesConsumed
263− | otherwise = bytes + 1
264− p = do _ <- getByteString (fromIntegral bytes)
265− _ <- getBits (fromIntegral bits) :: BitGet Word8
267− r = runGetIncremental (runBitGet p) `pushChunks` lbs
269− Fail remainingBS pos errMsg ->
270− and [ L.fromChunks [remainingBS] == L.drop expectedBytesConsumed lbs
271− , pos == expectedBytesConsumed
276−-- | number of bits required to write @v@
170+-- number of bits required to write @v@
277−bitreq :: (Num b, Num a, Bits a, Ord a) => a -> b
171+bitreq :: (Num b, Bits a, Ord a) => a -> b
278bitreq v = fromIntegral . head $ [ req | (req, top) <- bittable, v <= top ]
172bitreq v = fromIntegral . head $ [ req | (req, top) <- bittable, v <= top ]
280−bittable :: (Bits a, Num a) => [(Integer, a)]
174+bittable :: Bits a => [(Integer, a)]
281bittable = [ (fromIntegral x, (1 `shiftL` x) - 1) | x <- [1..64] ]
175bittable = [ (fromIntegral x, (1 `shiftL` x) - 1) | x <- [1..64] ]
283prop_bitreq :: W Word64 -> Property
177prop_bitreq :: W Word64 -> Property
284prop_bitreq (W w) = property $
178prop_bitreq (W w) = property $
285 ( w == 0 && bitreq w == 1 )
179 ( w == 0 && bitreq w == 1 )
286− || bitreq w == bitreq (w `shiftR` 1) + 1
180+ || bitreq w == (bitreq (w `shiftR` 1)) + 1
288prop_composite_case :: Bool -> W Word16 -> Property
182prop_composite_case :: Bool -> W Word16 -> Property
289prop_composite_case b (W w) = w < 0x8000 ==>
183prop_composite_case b (W w) = w < 0x8000 ==>
290 let p = do putBool b
184 let p = do putBool b
291− putWord16 15 w
185+ putWord16be 15 w
292 g = do v <- getBool
186 g = do v <- getBool
293 case v of
187 case v of
294− True -> getWord16 15
188+ True -> getWord16be 15
295 False -> do
189 False -> do
296 msb <- getWord8 7
190 msb <- getWord8 7
297 lsb <- getWord8 8
191 lsb <- getWord8 8
@@ -300,7 +194,8 @@ prop_composite_case b (W w) = w < 0x8000 ==>
300 w' = runGet (runBitGet g) lbs
194 w' = runGet (runBitGet g) lbs
301 in w == w'
195 in w == w'
303−prop_compare_put_with_naive :: (Bits a, BinaryBit a, Ord a, Num a) => W [a] -> Property
198+prop_compare_put_with_naive :: (Bits a, BinaryBit a, Ord a) => W [a] -> Property
304prop_compare_put_with_naive (W ws) = property $
199prop_compare_put_with_naive (W ws) = property $
305 let pn = mapM_ (\v -> naive_put (bitreq v) v) ws
200 let pn = mapM_ (\v -> naive_put (bitreq v) v) ws
306 p = mapM_ (\v -> putBits (bitreq v) v) ws
201 p = mapM_ (\v -> putBits (bitreq v) v) ws
@@ -317,7 +212,7 @@ prop_compare_get_with_naive (W ws) = property $
317 rn = runGet (runBitGet gn) lbs
212 rn = runGet (runBitGet gn) lbs
318 r = runGet (runBitGet g ) lbs
213 r = runGet (runBitGet g ) lbs
319 -- we must help our compiler to resolve the types of 'gn' and 'g'
214 -- we must help our compiler to resolve the types of 'gn' and 'g'
320− _types = rn == ws && r == ws
215+ types = rn == ws && r == ws
321 in rn == r
216 in rn == r
323-- | Write one bit at a time until the full word has been written
218-- | Write one bit at a time until the full word has been written
@@ -335,6 +230,27 @@ naive_get n0 =
335 True -> loop (n-1) ((acc `shiftL` 1) + 1)
230 True -> loop (n-1) ((acc `shiftL` 1) + 1)
336 in loop n0 0
231 in loop n0 0
234+prop_Word32_from_Word8_and_Word16 :: Word8 -> Word16 -> Property
235+prop_Word32_from_Word8_and_Word16 w8 w16 = property $
236+ let p = RWord32be 24
237+ w' = runGet (get p) lbs
240+ lbs = runPut (putWord8 w8 >> putWord16be w16)
241+ w0 = ((fromIntegral w8) `shiftL` 16) .|. fromIntegral w16
243+prop_Word32_from_2_Word16 :: Word16 -> Word16 -> Property
244+prop_Word32_from_2_Word16 w1 w2 = property $
245+ let p = RWord32be 32
246+ w' = runGet (get p) lbs
250+ w0 = ((fromIntegral w1) `shiftL` 16) .|. fromIntegral w2
338shrinker :: (Num a, Ord a, Bits a) => a -> [a]
254shrinker :: (Num a, Ord a, Bits a) => a -> [a]
339shrinker 0 = []
255shrinker 0 = []
340shrinker w = [ w `shiftR` 1 -- try to make everything roughly half size
256shrinker w = [ w `shiftR` 1 -- try to make everything roughly half size
@@ -348,16 +264,14 @@ shrinker w = [ w `shiftR` 1 -- try to make everything roughly half size
349data W a = W { unW :: a } deriving (Show, Eq, Ord)
265data W a = W { unW :: a } deriving (Show, Eq, Ord)
351−arbitraryW :: (Arbitrary (W a)) => Gen a
352−arbitraryW = unW <$> arbitrary
354−shrinkW :: (Arbitrary (W a)) => a -> [a]
355−shrinkW x = unW <$> shrink (W x)
357instance Arbitrary (W Bool) where
267instance Arbitrary (W Bool) where
358 arbitrary = W <$> arbitrary
268 arbitrary = W <$> arbitrary
359 shrink = map W <$> shrink . unW
269 shrink = map W <$> shrink . unW
271+instance (Arbitrary (W a)) => Arbitrary (W [a]) where
272+ arbitrary = W . map unW <$> arbitrary
273+ shrink = map (W . map unW) <$> mapM shrink . map W . unW
361instance Arbitrary (W Word8) where
275instance Arbitrary (W Word8) where
362 arbitrary = W <$> choose (minBound, maxBound)
276 arbitrary = W <$> choose (minBound, maxBound)
363 shrink = map W . shrinker . unW
277 shrink = map W . shrinker . unW
@@ -378,155 +292,11 @@ instance Arbitrary B.ByteString where
378 arbitrary = B.pack <$> arbitrary
292 arbitrary = B.pack <$> arbitrary
379 shrink bs = B.pack <$> shrink (B.unpack bs)
293 shrink bs = B.pack <$> shrink (B.unpack bs)
381−instance Arbitrary L.ByteString where
295+integralRandomR :: (Integral a, RandomGen g) => (a,a) -> g -> (a,g)
382− arbitrary = L.fromChunks <$> arbitrary
296+integralRandomR (a,b) g = case randomR (fromIntegral a :: Integer,
383− shrink bs = L.fromChunks <$> shrink (L.toChunks bs)
297+ fromIntegral b :: Integer) g of
384−
298+ (x,g) -> (fromIntegral x, g)
385−instance (Arbitrary (W a)) => Arbitrary (W [a]) where
386− arbitrary = W . map unW <$> arbitrary
387− shrink = map (W . map unW) <$> mapM shrink . map W . unW
389−instance (Arbitrary (W a), Arbitrary (W b)) => Arbitrary (W (a,b)) where
390− arbitrary = (W .) . (,) <$> arbitraryW <*> arbitraryW
391− shrink (W (a,b)) = (W .) . (,) <$> shrinkW a <*> shrinkW b
393−instance (Arbitrary (W a), Arbitrary (W b), Arbitrary (W c)) => Arbitrary (W (a,b,c)) where
394− arbitrary = ((W .) .) . (,,) <$> arbitraryW <*> arbitraryW <*> arbitraryW
395− shrink (W (a,b,c)) = ((W .) .) . (,,) <$> shrinkW a <*> shrinkW b <*> shrinkW c
403− | BS Int B.ByteString
404− | LBS Int L.ByteString
409−type Program = [Primitive]
411−instance Arbitrary Primitive where
414− let (maxBits, _) = (\w -> (finiteBitSize w, c undefined w)) undefined
415− bits <- choose (0, maxBits)
416− n <- choose (0, fromIntegral (2^bits-1))
419− [ Bool <$> arbitrary
424− , Skip <$> choose (0, 3000)
425− , do n <- choose (0,10)
427− return (BS n (B.pack cs))
428− , do n <- choose (0,10)
430− return (LBS n (L.pack cs))
434− let snk c x = map (\x' -> c (bitreq x') x') (shrinker x) in
436− Bool b -> if b then [Bool False] else []
438− W16 _ x -> snk W16 x
439− W32 _ x -> snk W32 x
440− W64 _ x -> snk W64 x
441− Skip x -> Skip <$> shrink x
442− BS _ bs -> let ws = B.unpack bs in map (\ws' -> BS (length ws') (B.pack ws')) (shrink ws)
443− LBS _ lbs -> let ws = L.unpack lbs in map (\ws' -> LBS (length ws') (L.pack ws')) (shrink ws)
446−instance Arbitrary BitOrder where
447− arbitrary = elements [BB, LB, LL, BL]
448− shrink LL = [BB,LB,BL]
454−prop_primitive :: Primitive -> Property
455−prop_primitive prim = property $
456− let p = putPrimitive prim
457− g = getPrimitive prim
458− lbs = runPut (runBitPut p)
459− r = runGet (runBitGet g) lbs
462−prop_program :: Program -> Property
463−prop_program program = property $
464− let p = mapM_ putPrimitive program
465− g = verifyProgram (8 * fromIntegral (L.length lbs)) program
466− lbs = runPut (runBitPut p)
467− r = runGet (runBitGet g) lbs
470−putPrimitive :: Primitive -> BitPut ()
474− W8 n x -> putWord8 n x
475− W16 n x -> putWord16 n x
476− W32 n x -> putWord32 n x
477− W64 n x -> putWord64 n x
478− Skip n -> skipBits n
479− BS _ bs -> putByteString bs
480− LBS _ lbs -> mapM_ putByteString (L.toChunks lbs)
481− IsEmpty -> return ()
483−getPrimitive :: Primitive -> BitGet Primitive
486− Bool _ -> Bool <$> getBool
487− W8 n _ -> W8 n <$> getWord8 n
488− W16 n _ -> W16 n <$> getWord16 n
489− W32 n _ -> W32 n <$> getWord32 n
490− W64 n _ -> W64 n <$> getWord64 n
491− Skip n -> skipBits n >> return (Skip n)
492− BS n _ -> BS n <$> getByteString n
493− LBS n _ -> LBS n <$> getLazyByteString n
494− IsEmpty -> isEmpty >> return IsEmpty
496−getPrimitiveSize :: Primitive -> Int
497−getPrimitiveSize p = case p of
508−verifyProgram :: Int -> Program -> BitGet Bool
509−verifyProgram totalLength ps0 = go 0 ps0
511− go _ [] = return True
515− let expected = pos == totalLength
517− if expected == actual
519− else error $ "isEmpty returned wrong value, expected "
520− ++ show expected ++ " but got " ++ show actual
521− _ -> check p (getPrimitive p) >> go (pos + getPrimitiveSize p) ps
526− else error $ "Roundtrip error: Expected "
527− ++ show x ++ " but got " ++ show y
530instance Random Word where
300instance Random Word where
531 randomR = integralRandomR
301 randomR = integralRandomR
532 random = randomR (minBound,maxBound)
302 random = randomR (minBound,maxBound)
@@ -546,4 +316,3 @@ instance Random Word32 where
546instance Random Word64 where
316instance Random Word64 where
547 randomR = integralRandomR
317 randomR = integralRandomR
548 random = randomR (minBound,maxBound)
318 random = randomR (minBound,maxBound)