Làm cách nào để yêu cầu QuickCheck chỉ tạo các chỉ mục danh sách hợp lệ cho một tham số?

Question

Giả sử tôi muốn viết một số xét nghiệm đơn vị cho hàm (!!).

my_prop xs n = ... 

Tôi muốn hạn chế n để chỉ lập chỉ mục hợp lệ và tôi biết tôi có thể làm một cái gì đó giống như

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

Nhưng điều này làm cho nó để đại đa số các trường hợp phát sinh là không hợp lệ và bị vứt bỏ . Có cách nào tôi có thể thiết lập mọi thứ để QuickCheck tạo danh sách xs trước tiên và sử dụng giá trị của nó để chỉ tạo các trường hợp hợp lệ là n?

Answer 1

Sử dụng forAll, bạn có thể chỉ định generator cho n phụ thuộc vào các đối số trước đó, ví dụ:

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

Answer 2

Bạn có thể làm một máy phát điện mà chỉ tạo ra chỉ số hợp lệ và viết tài sản của bạn như

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') 

loại bỏ lập luận Int.

Answer 3

Theo đề xuất của Daniel Wagner, một khả năng là tạo kiểu dữ liệu của riêng tôi và cung cấp cho trường hợp 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 là từ loại tùy chỉnh trong Test.QuickCheck.Modifiers để tạo danh sách không trống.