@@ -7,32 +7,24 @@ module Test.Hedgehog.Compat (
77 , forAllRender
88 )where
99
10−import Control.Monad.Trans.Class (MonadTrans(..))
11−
10+import Hedgehog (Gen)
1211import qualified Hedgehog.Gen as Gen
1312import qualified Hedgehog.Range as Range
1413import Hedgehog.Internal.Property
1514import Hedgehog.Internal.Source
16−import Hedgehog.Internal.Show
1715
1816(...) :: (c -> d) -> (a -> b -> c) -> a -> b -> d
1917(...) = (.) . (.)
2018{-# INLINE (...) #-}
2119
22−choose :: ( Monad m, Integral a ) => a -> a -> Gen.Gen m a
20+choose :: ( Integral a ) => a -> a -> Gen a
2321choose = Gen.integral ... Range.constant
2422
25−blindForAll :: Monad m => Gen.Gen m a -> Test m a
26−blindForAll = Test . lift . lift
23+blindForAll :: Monad m => Gen a -> PropertyT m a
24+blindForAll = forAllWith (const "blind")
2725
28−semiBlindForAll :: (Monad m, Show a, HasCallStack) => Gen.Gen m a -> Test m a
29−semiBlindForAll gen = do
30− x <- Test . lift $ lift gen
31− withFrozenCallStack $ annotate (showPretty x)
32− return x
26+semiBlindForAll :: (Monad m, Show a, HasCallStack) => Gen a -> PropertyT m a
27+semiBlindForAll = forAllWith (const "blind")
3328
34−forAllRender :: (Monad m, HasCallStack) => ( a -> String ) -> Gen.Gen m a -> Test m a
35−forAllRender render gen = do
36− x <- Test . lift $ lift gen
37− withFrozenCallStack $ footnote (render x)
38− return x
29+forAllRender :: (Monad m, HasCallStack) => ( a -> String ) -> Gen a -> PropertyT m a
30+forAllRender render gen = forAllWith render gen