khibino/haskell-relational-record
1
1
Fork 0
mirror of https://github.com/khibino/haskell-relational-record.git synced 2025-01-06 04:16:21 +03:00
Code Issues Projects Releases Wiki Activity

switch to rangeN.

Browse Source
This commit is contained in:
Kei Hibino 2016-12-19 18:15:49 +09:00
parent 9d444c89ad
commit 9ec6c3726b
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
rr-quickcheck/src/Test/Relational/QuickCheck/Arbitrary.hs
View File

@ -85,12 +85,15 @@ newtype D a =
D { unD :: a }
deriving Show
rangeN :: (Num a, Enum a) => a -> Gen a
rangeN n = elements [-n .. n]
genVarExpr :: Gen (Selector a) -> Gen (VarExpr a)
genVarExpr gs =
frequency
[ (3, Column <$> gs)
, (1, VLeft <$> rec' <*> arbitrary <*> elements [-10 .. 10])
, (1, VRight <$> elements [-10 .. 10] <*> arbitrary <*> rec')
, (1, VLeft <$> rec' <*> arbitrary <*> rangeN 5)
, (1, VRight <$> rangeN 5 <*> arbitrary <*> rec')
]
where
rec' = genVarExpr gs
@ -236,21 +239,18 @@ predHask r = d where
newtype Ranged a = Ranged { runRanged :: [a] }
deriving (Eq, Show)
range5 :: (Num a, Enum a) => Gen a
range5 = elements [-5 .. 5]
instance Arbitrary (Ranged A) where
arbitrary =
(Ranged <$>) . listOf $
A
<$> range5
<*> range5
<*> range5
<$> rangeN 5
<*> rangeN 5
<*> rangeN 5
instance Arbitrary (Ranged B) where
arbitrary =
(Ranged <$>) . listOf $
B
<$> range5
<*> range5
<*> range5
<$> rangeN 5
<*> rangeN 5
<*> rangeN 5