Как сообщить QuickCheck о создании только допустимых индексов списка для параметра?

Скажем, я хочу написать некоторые модульные тесты для функции (!!).

my_prop xs n = ...

Я хочу ограничить n только действительными индексами, и я знаю, что могу сделать что-то вроде

my_prop xs n = (not.null) (drop n xs) ==> ...

Но это делает так, что подавляющее большинство сгенерированных случаев являются недействительными и отбрасываются. Есть ли способ, который я могу настроить, чтобы QuickCheck сначала генерировал список xs и использует его значение для генерации только допустимых случаев n?

Ответ 1

Используя forAll, вы можете указать генератор для n, который зависит от предыдущих аргументов, например

my_prop (NonEmpty xs) = forAll (choose (0, length xs - 1)) $ \n -> ...

Ответ 2

Вы можете создать генератор, который создает только действительные индексы и записывает ваше свойство, например

import Test.QuickCheck
import Test.QuickCheck.Gen
import System.Random

indices :: [a] -> Gen Int
indices xs = MkGen $ \sg _ -> fst $ randomR (0, length xs - 1) sg

my_prop :: [Char] -> Property
my_prop xs = not (null xs) ==> forAll (indices xs) (\i -> xs !! i /= '0')

устранение аргумента Int.

Ответ 3

Как было предложено Дэниелом Вагнером, одна возможность создает мой собственный тип данных и дает ему экземпляр Arbitrary.

data ListAndIndex a = ListAndIndex [a] Int deriving (Show)

instance Arbitrary a => Arbitrary (ListAndIndex a) where
   arbitrary = do
     (NonEmpty xs) <- arbitrary
     n  <- elements [0..(length xs - 1)]
     return $ ListAndIndex xs n

NonEmpty относится к пользовательскому типу в Test.QuickCheck.Modifiers для генерации непустых списков.