Merge pull request #253 from felixmulder/topic/add-classifications

Add classifications to properties

hedgehog-example/src/Test/Example/Coverage.hs

@@ -0,0 +1,40 @@
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE TemplateHaskell #-}
+module Test.Example.Coverage (
+    tests
+  ) where
+
+import           Control.Concurrent (threadDelay)
+
+import           Data.Foldable (for_)
+
+import           Hedgehog
+import qualified Hedgehog.Gen as Gen
+import qualified Hedgehog.Range as Range
+
+prop_classify :: Property
+prop_classify =
+  withTests 1 . property $ do
+    for_ [1 :: Int ..100] $ \a -> do
+      classify "small number" $ a < 50
+      classify "big number" $ a >= 50
+
+prop_cover_number :: Property
+prop_cover_number =
+  property $ do
+    number <- forAll (Gen.int $ Range.linear 1 100)
+    evalIO $ threadDelay 20000
+    cover 50 "small number" $ number < 50
+    cover 50 "medium number" $ number >= 20
+    cover 50 "big number" $ number >= 50
+
+prop_cover_bool :: Property
+prop_cover_bool =
+  property $ do
+    match <- forAll Gen.bool
+    cover 30 "True" match
+    cover 30 "False" $ not match
+
+tests :: IO Bool
+tests =
+  checkParallel $$(discover)

hedgehog/src/Hedgehog.hs

@@ -56,6 +56,8 @@ module Hedgehog (
 
   , forAll
   , forAllWith
+  , classify
+  , cover
   , discard
 
   , check
@@ -151,8 +153,9 @@ import           Hedgehog.Internal.Distributive (Distributive(..))
 import           Hedgehog.Internal.Gen (Gen, GenT, MonadGen(..))
 import           Hedgehog.Internal.HTraversable (HTraversable(..))
 import           Hedgehog.Internal.Opaque (Opaque(..))
-import           Hedgehog.Internal.Property (assert, diff, annotate, annotateShow)
-import           Hedgehog.Internal.Property ((===), (/==))
+import           Hedgehog.Internal.Property (annotate, annotateShow)
+import           Hedgehog.Internal.Property (assert, diff, (===), (/==))
+import           Hedgehog.Internal.Property (classify, cover)
 import           Hedgehog.Internal.Property (discard, failure, success)
 import           Hedgehog.Internal.Property (DiscardLimit, withDiscards)
 import           Hedgehog.Internal.Property (eval, evalM, evalIO)