@@ -1,532 +1,142 @@
1−{-# LANGUAGE CPP, FlexibleInstances, FlexibleContexts, TupleSections #-}
1+module Main where
22
3−module Main ( main ) where
4−
5−import Data.Binary ( Binary(..) )
6−import Data.Binary.Get ( runGet, runGetIncremental, pushChunks, Decoder(..) )
3+import Data.Binary ( encode )
4+import Data.Binary.Get ( runGet )
75import Data.Binary.Put ( runPut )
86
9−import Data.Binary.Bits
10−import Data.Binary.Bits.Get
11−import Data.Binary.Bits.Put
12−import Data.Binary.Bits.BitOrder
13−
14−import qualified Data.ByteString as B
15−import qualified Data.ByteString.Lazy as L
7+import Bits
168
17−import Control.Applicative
189import Data.Bits
1910import Data.Word
2011import Foreign.Storable
21−import Data.Traversable (traverse)
22−import Data.Foldable (traverse_)
23−
24−import Test.Framework.Providers.QuickCheck2 ( testProperty )
25−import Test.Framework.Runners.Console ( defaultMain )
26−import Test.Framework ( Test, testGroup )
27−import Test.QuickCheck
28−
29−main :: IO ()
30−main = defaultMain tests
31−
32−tests :: [Test]
33−tests =
34− [ testGroup "Internal test functions"
35− [ testProperty "prop_bitreq" prop_bitreq ]
36−
37− , testGroup "Custom test cases"
38− [ testProperty "prop_composite_case" prop_composite_case ]
39−
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)
44− ]
45−
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
51− ]
52−
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
58− ]
59−
60− , testGroup "Fail"
61− [ testProperty "monadic fail" prop_fail ]
62−
63− , testGroup "prop_bitput_with_get_from_binary"
64− [ testProperty "Word8" (prop_bitput_with_get_from_binary :: W [Word8] -> Property)
65− , testProperty "Word16" (prop_bitput_with_get_from_binary :: W [Word16] -> Property)
66− , testProperty "Word32" (prop_bitput_with_get_from_binary :: W [Word32] -> Property)
67− , testProperty "Word64" (prop_bitput_with_get_from_binary :: W [Word64] -> Property)
68− ]
69−
70− , testGroup "prop_bitget_with_put_from_binary"
71− [ testProperty "Word8" (prop_bitget_with_put_from_binary :: W [Word8] -> Property)
72− , testProperty "Word16" (prop_bitget_with_put_from_binary :: W [Word16] -> Property)
73− , testProperty "Word32" (prop_bitget_with_put_from_binary :: W [Word32] -> Property)
74− , testProperty "Word64" (prop_bitget_with_put_from_binary :: W [Word64] -> Property)
75− ]
76−
77− , testGroup "prop_compare_put_with_naive"
78− [ testProperty "Word8" (prop_compare_put_with_naive :: W [Word8] -> Property)
79− , testProperty "Word16" (prop_compare_put_with_naive :: W [Word16] -> Property)
80− , testProperty "Word32" (prop_compare_put_with_naive :: W [Word32] -> Property)
81− , testProperty "Word64" (prop_compare_put_with_naive :: W [Word64] -> Property)
82− ]
83−
84− , testGroup "prop_compare_get_with_naive"
85− [ testProperty "Word8" (prop_compare_get_with_naive:: W [Word8] -> Property)
86− , testProperty "Word16" (prop_compare_get_with_naive:: W [Word16] -> Property)
87− , testProperty "Word32" (prop_compare_get_with_naive:: W [Word32] -> Property)
88− , testProperty "Word64" (prop_compare_get_with_naive:: W [Word64] -> Property)
89− ]
90−
91− , testGroup "prop_put_with_bitreq"
92− [ testProperty "Word8" (prop_putget_with_bitreq :: BitOrder -> W Word8 -> Property)
93− , testProperty "Word16" (prop_putget_with_bitreq :: BitOrder -> W Word16 -> Property)
94− , testProperty "Word32" (prop_putget_with_bitreq :: BitOrder -> W Word32 -> Property)
95− , testProperty "Word64" (prop_putget_with_bitreq :: BitOrder -> W Word64 -> Property)
96− ]
97−
98− , testGroup "prop_putget_list_simple"
99− [ testProperty "Bool" (prop_putget_list_simple :: W [Bool] -> Property)
100− , testProperty "Word8" (prop_putget_list_simple :: W [Word8] -> Property)
101− , testProperty "Word16" (prop_putget_list_simple :: W [Word16] -> Property)
102− , testProperty "Word32" (prop_putget_list_simple :: W [Word32] -> Property)
103− , testProperty "Word64" (prop_putget_list_simple :: W [Word64] -> Property)
104− ]
12+import System.Random
10513
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)
111− ]
14+import BitsGet
15+import BitsPut
11216
113− , testGroup "prop_putget_list_with_bitreq"
114− [ testProperty "Word8" (prop_putget_list_with_bitreq :: W [Word8] -> Property)
115− , testProperty "Word16" (prop_putget_list_with_bitreq :: W [Word16] -> Property)
116− , testProperty "Word32" (prop_putget_list_with_bitreq :: W [Word32] -> Property)
117− , testProperty "Word64" (prop_putget_list_with_bitreq :: W [Word64] -> Property)
118− ]
119− , testGroup "prop_bitget_bytestring_interspersed"
120− [ testProperty "Word8" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word8 -> [B.ByteString] -> Property)
121− , testProperty "Word16" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word16 -> [B.ByteString] -> Property)
122− , testProperty "Word32" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word32 -> [B.ByteString] -> Property)
123− , testProperty "Word64" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word64 -> [B.ByteString] -> Property)
124− ]
125− , testGroup "Simulate programs"
126− [ testProperty "primitive" prop_primitive
127− , testProperty "many primitives in sequence" prop_program
128− ]
129− ]
130−
131−prop_isEmptyOfEmptyEmpty :: Bool
132−prop_isEmptyOfEmptyEmpty = runGet (runBitGet isEmpty) L.empty
133−
134−prop_isEmptyOfNonEmptyEmpty :: L.ByteString -> Property
135−prop_isEmptyOfNonEmptyEmpty bs =
136− not (L.null bs) ==> not (runGet (runBitGet isEmpty) bs)
137−
138−prop_isEmptyOfConsumedEmpty :: L.ByteString -> Property
139−prop_isEmptyOfConsumedEmpty bs =
140− not (L.null bs) ==>
141− runGet (runBitGet (getByteString n >> isEmpty)) bs
142− where n = fromIntegral $ L.length bs
143−
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)
148−
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)
154−
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)
17+import Test.QuickCheck
16018
161−prop_getByteString_negative :: Int -> Property
162−prop_getByteString_negative n =
163− n < 1 ==>
164− runGet (runBitGet (getByteString n)) L.empty == B.empty
19+main = do
20+ quickCheck prop_Bools
21+ quickCheck prop_SimpleCase
16522
166−prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property
167−prop_putByteString_getByteString bo bs = property $ bs' == bs
168− where
169− n = B.length bs
170− w = runPut (runBitPut (withBitOrder bo (putByteString bs)))
171− bs' = runGet (runBitGet (withBitOrder bo (getByteString n))) w
23+ quickCheck (prop_putget_with_bitreq :: Word8 -> Property)
24+ quickCheck (prop_putget_with_bitreq :: Word16 -> Property)
17225
173−prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property
174−prop_putByteString_getByteString_many bo bs = property $ bs' == bs
175− where
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
26+ quickCheck (prop_putget_list_simple :: [Bool] -> Property)
27+ quickCheck (prop_putget_list_simple :: [Word8] -> Property)
28+ quickCheck (prop_putget_list_simple :: [Word16] -> Property)
17929
30+ quickCheck (prop_putget_list_with_bitreq :: [Word8] -> Property)
31+ quickCheck (prop_putget_list_with_bitreq :: [Word16] -> Property)
18032
181−prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => BitOrder -> W a -> Property
182−prop_putget_with_bitreq bo (W w) = property $
33+ -- these tests use the R structure
34+ --
35+ -- quickCheck prop_Word32_from_2_Word16
36+ -- quickCheck prop_Word32_from_Word8_and_Word16
37+
38+prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => a -> Property
39+prop_putget_with_bitreq w = property $
18340 -- write all words with as many bits as it's required
18441 let p = putBits (bitreq w) w
18542 g = getBits (bitreq w)
186− lbs = runPut (runBitPut (withBitOrder bo p))
187− w' = runGet (runBitGet (withBitOrder bo g)) lbs
43+ lbs = runPut (runBitPut p)
44+ w' = runGet (runBitGetSimple g) lbs
18845 in w == w'
18946
190−-- | Write a list of items. Each item is written with the maximum amount of
191−-- bits, i.e. 8 for Word8, 16 for Word16, etc.
192−prop_putget_list_simple :: (BinaryBit a, Eq a, Storable a) => W [a] -> Property
193−prop_putget_list_simple (W ws) = property $
47+prop_putget_list_simple :: (BinaryBit a, Eq a, Storable a) => [a] -> Property
48+prop_putget_list_simple ws = property $
19449 let s = sizeOf (head ws) * 8
195− p = mapM_ (putBits s) ws
50+ p = mapM_ (\v -> putBits s v) ws
19651 g = mapM (const (getBits s)) ws
19752 lbs = runPut (runBitPut p)
198− ws' = runGet (runBitGet g) lbs
53+ ws' = runGet (runBitGetSimple g) lbs
19954 in ws == ws'
20055
201−-- | Write a list of items. Each item is written with exactly as many bits
202−-- as required. Then read it back.
203−prop_putget_list_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => W [a] -> Property
204−prop_putget_list_with_bitreq (W ws) = property $
56+prop_putget_list_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => [a] -> Property
57+prop_putget_list_with_bitreq ws = property $
20558 -- write all words with as many bits as it's required
20659 let p = mapM_ (\v -> putBits (bitreq v) v) ws
20760 g = mapM getBits bitlist
20861 lbs = runPut (runBitPut p)
209− ws' = runGet (runBitGet g) lbs
62+ ws' = runGet (runBitGetSimple g) lbs
21063 in ws == ws'
21164 where
21265 bitlist = map bitreq ws
21366
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
222− in ts == ts'
223− where
224− bitlist = map (\(a,b,c) -> (bitreq a, bitreq b, bitreq c)) ts
225−
226−-- | Write bits using this library, and read them back using the binary
227−-- library.
228−prop_bitput_with_get_from_binary :: (BinaryBit a, Binary a, Storable a, Eq a) => W [a] -> Property
229−prop_bitput_with_get_from_binary (W ws) = property $
230− let s = sizeOf (head ws) * 8
231− p = mapM_ (putBits s) ws
232− g = mapM (const get) ws
233− lbs = runPut (runBitPut p)
234− ws' = runGet g lbs
235− in ws == ws'
236−
237−-- | Write bits using the binary library, and read them back using this
238−-- library.
239−prop_bitget_with_put_from_binary :: (BinaryBit a, Binary a, Storable a, Eq a) => W [a] -> Property
240−prop_bitget_with_put_from_binary (W ws) = property $
241− let s = sizeOf (head ws) * 8
242− p = mapM_ put ws
243− g = mapM (const (getBits s)) ws
244− lbs = runPut p
245− ws' = runGet (runBitGet g) lbs
246− in ws == ws'
247−
248−-- | 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
250−prop_bitget_bytestring_interspersed bo (W ws) bss = property $
251− let p = mapM_ (\bs -> putBits (bitreq ws) ws >> putByteString bs) bss
252− g = mapM (\bs -> (,) <$> getBits (bitreq ws) <*> getByteString (B.length bs)) bss
253− lbs = runPut (runBitPut (withBitOrder bo p))
254− r = runGet (runBitGet (withBitOrder bo g)) lbs
255− in map (ws,) bss == r
256−
257−-- | Test failing.
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
262− | bits == 0 = bytes
263− | otherwise = bytes + 1
264− p = do _ <- getByteString (fromIntegral bytes)
265− _ <- getBits (fromIntegral bits) :: BitGet Word8
266− fail errMsg0
267− r = runGetIncremental (runBitGet p) `pushChunks` lbs
268− in case r of
269− Fail remainingBS pos errMsg ->
270− and [ L.fromChunks [remainingBS] == L.drop expectedBytesConsumed lbs
271− , pos == expectedBytesConsumed
272− , errMsg == errMsg0
273− ]
274− _ -> False
275−
276−-- | number of bits required to write @v@
277−bitreq :: (Num b, Num a, Bits a, Ord a) => a -> b
67+-- number of bits required to write 'v'
68+bitreq :: (Num b, Bits a, Ord a) => a -> b
27869bitreq v = fromIntegral . head $ [ req | (req, top) <- bittable, v <= top ]
27970
280−bittable :: (Bits a, Num a) => [(Integer, a)]
71+bittable :: Bits a => [(Integer, a)]
28172bittable = [ (fromIntegral x, (1 `shiftL` x) - 1) | x <- [1..64] ]
28273
283−prop_bitreq :: W Word64 -> Property
284−prop_bitreq (W w) = property $
285− ( w == 0 && bitreq w == 1 )
286− || bitreq w == bitreq (w `shiftR` 1) + 1
74+prop_Bools :: [Bool] -> Property
75+prop_Bools bs = property $
76+ let p = sequence . replicate (length bs) $ getBool
77+ Right bs' = runGet (runBitGet p) lbs
78+ in bs == bs'
79+ where lbs = runPut $ runBitPut (mapM_ putBool bs)
28780
288−prop_composite_case :: Bool -> W Word16 -> Property
289−prop_composite_case b (W w) = w < 0x8000 ==>
81+prop_SimpleCase :: Bool -> Word16 -> Property
82+prop_SimpleCase b w = w < 0x8000 ==>
29083 let p = do putBool b
291− putWord16 15 w
84+ putWord16be 15 w
29285 g = do v <- getBool
29386 case v of
294− True -> getWord16 15
87+ True -> getWord16be 15
29588 False -> do
29689 msb <- getWord8 7
29790 lsb <- getWord8 8
29891 return ((fromIntegral msb `shiftL` 8) .|. fromIntegral lsb)
29992 lbs = runPut (runBitPut p)
300− w' = runGet (runBitGet g) lbs
93+ w' = runGet (runBitGetSimple g) lbs
30194 in w == w'
302−
303−prop_compare_put_with_naive :: (Bits a, BinaryBit a, Ord a, Num a) => W [a] -> Property
304−prop_compare_put_with_naive (W ws) = property $
305− let pn = mapM_ (\v -> naive_put (bitreq v) v) ws
306− p = mapM_ (\v -> putBits (bitreq v) v) ws
307− lbs_n = runPut (runBitPut pn)
308− lbs = runPut (runBitPut p)
309− in lbs_n == lbs
310−
311−prop_compare_get_with_naive :: (Bits a, BinaryBit a, Ord a, Num a) => W [a] -> Property
312−prop_compare_get_with_naive (W ws) = property $
313− let gn = mapM (\v -> naive_get (bitreq v)) ws
314− g = mapM (\v -> getBits (bitreq v)) ws
315− p = mapM_ (\v -> naive_put (bitreq v) v) ws
316− lbs = runPut (runBitPut p)
317− rn = runGet (runBitGet gn) lbs
318− r = runGet (runBitGet g ) lbs
319− -- we must help our compiler to resolve the types of 'gn' and 'g'
320− _types = rn == ws && r == ws
321− in rn == r
322−
323−-- | Write one bit at a time until the full word has been written
324−naive_put :: (Bits a) => Int -> a -> BitPut ()
325−naive_put n w = mapM_ (\b -> putBool (testBit w b)) [n-1,n-2..0]
326−
327−-- | Read one bit at a time until we've reconstructed the whole word
328−naive_get :: (Bits a, Num a) => Int -> BitGet a
329−naive_get n0 =
330− let loop 0 acc = return acc
331− loop n acc = do
332− b <- getBool
333− case b of
334− False -> loop (n-1) (acc `shiftL` 1)
335− True -> loop (n-1) ((acc `shiftL` 1) + 1)
336− in loop n0 0
337−
338−shrinker :: (Num a, Ord a, Bits a) => a -> [a]
339−shrinker 0 = []
340−shrinker w = [ w `shiftR` 1 -- try to make everything roughly half size
341− ] ++ [ w' -- flip bits to zero, left->right
342− | m <- [n, n-1..1]
343− , let w' = w `clearBit` m
344− , w /= w'
345− ] ++ [w-1] -- just make it a little smaller
34695 where
347− n = bitreq w
348−
349−data W a = W { unW :: a } deriving (Show, Eq, Ord)
350−
351−arbitraryW :: (Arbitrary (W a)) => Gen a
352−arbitraryW = unW <$> arbitrary
353−
354−shrinkW :: (Arbitrary (W a)) => a -> [a]
355−shrinkW x = unW <$> shrink (W x)
356−
357−instance Arbitrary (W Bool) where
358− arbitrary = W <$> arbitrary
359− shrink = map W <$> shrink . unW
360−
361−instance Arbitrary (W Word8) where
362− arbitrary = W <$> choose (minBound, maxBound)
363− shrink = map W . shrinker . unW
364−
365−instance Arbitrary (W Word16) where
366− arbitrary = W <$> choose (minBound, maxBound)
367− shrink = map W . shrinker . unW
368−
369−instance Arbitrary (W Word32) where
370− arbitrary = W <$> choose (minBound, maxBound)
371− shrink = map W . shrinker . unW
372−
373−instance Arbitrary (W Word64) where
374− arbitrary = W <$> choose (minBound, maxBound)
375− shrink = map W . shrinker . unW
37696
377−instance Arbitrary B.ByteString where
378− arbitrary = B.pack <$> arbitrary
379− shrink bs = B.pack <$> shrink (B.unpack bs)
380−
381−instance Arbitrary L.ByteString where
382− arbitrary = L.fromChunks <$> arbitrary
383− shrink bs = L.fromChunks <$> shrink (L.toChunks bs)
384−
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
388−
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
392−
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
396−
397−data Primitive
398− = Bool Bool
399− | W8 Int Word8
400− | W16 Int Word16
401− | W32 Int Word32
402− | W64 Int Word64
403− | BS Int B.ByteString
404− | LBS Int L.ByteString
405− | Skip Int
406− | IsEmpty
407− deriving (Eq, Show)
408−
409−type Program = [Primitive]
410−
411−instance Arbitrary Primitive where
412− arbitrary = do
413− let gen c = do
414− let (maxBits, _) = (\w -> (finiteBitSize w, c undefined w)) undefined
415− bits <- choose (0, maxBits)
416− n <- choose (0, fromIntegral (2^bits-1))
417− return (c bits n)
418− oneof
419− [ Bool <$> arbitrary
420− , gen W8
421− , gen W16
422− , gen W32
423− , gen W64
424− , Skip <$> choose (0, 3000)
425− , do n <- choose (0,10)
426− cs <- vector n
427− return (BS n (B.pack cs))
428− , do n <- choose (0,10)
429− cs <- vector n
430− return (LBS n (L.pack cs))
431− , return IsEmpty
432− ]
433− shrink p =
434− let snk c x = map (\x' -> c (bitreq x') x') (shrinker x) in
435− case p of
436− Bool b -> if b then [Bool False] else []
437− W8 _ x -> snk W8 x
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)
444− IsEmpty -> []
445−
446−instance Arbitrary BitOrder where
447− arbitrary = elements [BB, LB, LL, BL]
448− shrink LL = [BB,LB,BL]
449− shrink BL = [BB,LB]
450− shrink LB = [BB]
451− shrink BB = []
45297
98+{-
99+prop_Word32_from_Word8_and_Word16 :: Word8 -> Word16 -> Property
100+prop_Word32_from_Word8_and_Word16 w8 w16 = property $
101+ let p = RWord32be 24
102+ w' = runGet (get p) lbs
103+ in w0 == w'
104+ where
105+ lbs = runPut (putWord8 w8 >> putWord16be w16)
106+ w0 = ((fromIntegral w8) `shiftL` 16) .|. fromIntegral w16
107+
108+prop_Word32_from_2_Word16 :: Word16 -> Word16 -> Property
109+prop_Word32_from_2_Word16 w1 w2 = property $
110+ let p = RWord32be 32
111+ w' = runGet (get p) lbs
112+ in w0 == w'
113+ where
114+ lbs = encode w0
115+ w0 = ((fromIntegral w1) `shiftL` 16) .|. fromIntegral w2
116+-}
453117
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
460− in r == prim
118+instance Arbitrary Word8 where
119+ arbitrary = choose (minBound, maxBound)
120+ shrink 0 = []
121+ shrink n = [ n - 1 ]
461122
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
468− in r
123+instance Arbitrary Word16 where
124+ arbitrary = choose (minBound, maxBound)
125+ shrink 0 = []
126+ shrink n = [ n - 10000, n - 1000, n - 100, n - 1 ]
469127
470−putPrimitive :: Primitive -> BitPut ()
471−putPrimitive p =
472− case p of
473− Bool b -> putBool b
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 ()
128+instance Arbitrary Word32 where
129+ arbitrary = choose (minBound, maxBound)
482130
483−getPrimitive :: Primitive -> BitGet Primitive
484−getPrimitive p =
485− case p of
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
131+instance Arbitrary Word64 where
132+ arbitrary = choose (minBound, maxBound)
495133
496−getPrimitiveSize :: Primitive -> Int
497−getPrimitiveSize p = case p of
498− Bool _ -> 1
499− W8 n _ -> n
500− W16 n _ -> n
501− W32 n _ -> n
502− W64 n _ -> n
503− Skip n -> n
504− BS n _ -> n*8
505− LBS n _ -> n*8
506− IsEmpty -> 0
507134
508−verifyProgram :: Int -> Program -> BitGet Bool
509−verifyProgram totalLength ps0 = go 0 ps0
510− where
511− go _ [] = return True
512− go pos (p:ps) =
513− case p of
514− IsEmpty -> do
515− let expected = pos == totalLength
516− actual <- isEmpty
517− if expected == actual
518− then go pos ps
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
522− check x g = do
523− y <- g
524− if x == y
525− then return ()
526− else error $ "Roundtrip error: Expected "
527− ++ show x ++ " but got " ++ show y
135+integralRandomR :: (Integral a, RandomGen g) => (a,a) -> g -> (a,g)
136+integralRandomR (a,b) g = case randomR (fromIntegral a :: Integer,
137+ fromIntegral b :: Integer) g of
138+ (x,g) -> (fromIntegral x, g)
528139
529−{-
530140instance Random Word where
531141 randomR = integralRandomR
532142 random = randomR (minBound,maxBound)
@@ -546,4 +156,3 @@ instance Random Word32 where
546156instance Random Word64 where
547157 randomR = integralRandomR
548158 random = randomR (minBound,maxBound)
549−-}