Для простых свойств вашего ввода вы можете создать новый тип с соответствующим экземпляром Arbitrary
, который их захватывает.Итак:
{-# LANGUAGE AllowAmbiguousTypes #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE KindSignatures #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeApplications #-}
import Data.Proxy
import GHC.TypeLits
import Test.QuickCheck
newtype Range (m :: Nat) (n :: Nat) a = Range { getVal :: a }
deriving (Eq, Ord, Read, Show, Num, Real, Enum, Integral)
numVal :: forall n a. (KnownNat n, Num a) => a
numVal = fromInteger (natVal @n Proxy)
instance (KnownNat m, KnownNat n, Arbitrary a, Integral a) => Arbitrary (Range m n a) where
arbitrary = fromInteger <$> choose (numVal @m, numVal @n)
shrink hi = go (numVal @m) where
go lo | lo == hi = [] | otherwise = lo : go ((lo+hi+1)`div`2) -- overflow? what's that? lmao
whatAge :: Range 1 30 Int -> Range 1 40 Int -> Range 1 50 Int -> Range 1 60 Int -> Bool
whatAge (Range age1) (Range age2) (Range age3) (Range age4)
= age1 + age2 + age3 + age4 == 5
В ghci:
> verboseCheck whatAge
Failed:
Range {getVal = 17}
Range {getVal = 29}
Range {getVal = 3}
Range {getVal = 16}
Failed:
Range {getVal = 1}
Range {getVal = 29}
Range {getVal = 3}
Range {getVal = 16}
Failed:
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 3}
Range {getVal = 16}
Failed:
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 16}
Failed:
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 1}
*** Failed! Falsifiable (after 1 test and 4 shrinks):
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 1}
Range {getVal = 1}
Для более сложных свойств, где непонятно, как напрямую создать случайное значение, которое удовлетворяет свойству, вы можете использовать QuickCheck's (==>)
оператор.Например, для проверки диапазона, как указано выше:
> verboseCheck (\x -> (1 <= x && x <= 30) ==> x*2 < 60)
Skipped (precondition false):
0
Passed:
1
*** Failed! Falsifiable (after 33 tests):
30
Чтобы выполнить ровно 200 тестов, вы можете вызвать quickCheckWith
, чтобы сделать один тест 200 раз;или вы можете напрямую generate
проверить результаты, позвонив в вашу собственность по arbitrary
вручную.