Add property tests for genNonEmptyDisjointSet.

cardano-foundation · May 7, 2024 · 247442a · 247442a
1 parent ce5307f
commit 247442a
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/lib/test-utils/test/Test/QuickCheck/ExtraSpec.hs b/lib/test-utils/test/Test/QuickCheck/ExtraSpec.hs
@@ -87,10 +87,12 @@ import Test.QuickCheck
     , scale
     , shrinkIntegral
     , within
+    , (=/=)
     , (===)
     )
 import Test.QuickCheck.Extra
     ( Pretty (..)
+    , genNonEmptyDisjointSet
     , genShrinkSequence
     , genericRoundRobinShrink
     , getDisjointPair
@@ -191,6 +193,17 @@ spec = describe "Test.QuickCheck.ExtraSpec" $ do
                 prop_selectMapEntries_union
                     @Int @Int & property
 
+    describe "Generating and shrinking sets" $ do
+
+        describe "Generation" $ do
+
+            it "prop_genNonEmptyDisjointSet_disjoint" $
+                prop_genNonEmptyDisjointSet_disjoint
+                    @Int & property
+            it "prop_genNonEmptyDisjointSet_nonEmpty" $
+                prop_genNonEmptyDisjointSet_nonEmpty
+                    @Int & property
+
     describe "Generating and shrinking associative maps" $ do
 
         describe "Shrinking" $ do
@@ -575,6 +588,34 @@ prop_partitionList_LT (PartitionListData (x, y) as) =
     x' = max 0 x
     y' = max 1 (max y x')
 
+--------------------------------------------------------------------------------
+-- Generating sets
+--------------------------------------------------------------------------------
+
+prop_genNonEmptyDisjointSet_disjoint
+    :: (Arbitrary a, Ord a, Show a)
+    => Set a
+    -> Property
+prop_genNonEmptyDisjointSet_disjoint set1 =
+    forAll (genNonEmptyDisjointSet arbitrary set1) $ \set2 ->
+        Set.intersection set1 set2 === Set.empty
+        & cover 10
+            (Set.size set1 >= 10 && Set.size set2 >= 10)
+            "Set.size set1 >= 10 && Set.size set2 >= 10"
+        & checkCoverage
+
+prop_genNonEmptyDisjointSet_nonEmpty
+    :: (Arbitrary a, Ord a, Show a)
+    => Set a
+    -> Property
+prop_genNonEmptyDisjointSet_nonEmpty set1 =
+    forAll (genNonEmptyDisjointSet arbitrary set1) $ \set2 ->
+        Set.size set2 =/= 0
+        & cover 10
+            (Set.size set1 >= 10 && Set.size set2 >= 10)
+            "Set.size set1 >= 10 && Set.size set2 >= 10"
+        & checkCoverage
+
 --------------------------------------------------------------------------------
 -- Selecting map entries (one at a time)
 --------------------------------------------------------------------------------