sized-types-0.5.1: test/QC/QC.hs

-- Copy this module if you need Quick Check.
{-# OPTIONS_GHC -fno-warn-orphans #-}
module QC.QC where

import           Data.Ix
import           Data.Sized.Fin
import           Data.Sized.Matrix

import qualified Test.QuickCheck as QC

instance (SingI n) => QC.Arbitrary (Fin n) where
    arbitrary = QC.elements [minBound .. maxBound]

instance (QC.Arbitrary ix, Bounded ix, Ix ix, QC.Arbitrary a) => QC.Arbitrary (Matrix ix a) where
    arbitrary = f $ \ ixs -> do
          elems <- sequence [ QC.arbitrary | _ <- ixs ]
          return $ matrix elems
         where f :: (Bounded ix, Ix ix) => ([ix] -> m (Matrix ix a)) -> m (Matrix ix a)
               f fn = fn (allIndices (undefined :: Matrix ix a))