found another bug!

[no ci]
IntersectMBO · May 7, 2024 · 1a1c35e · 1a1c35e
1 parent b4b1e9c
commit 1a1c35e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/libs/constrained-generators/src/Constrained/Base.hs b/libs/constrained-generators/src/Constrained/Base.hs
@@ -2773,6 +2773,10 @@ guardSumSpec s@(SumSpec sa sb)
 
 data SumSpec fn a b = SumSpec (Specification fn a) (Specification fn b)
 
+instance (Arbitrary (Specification fn a), Arbitrary (Specification fn b)) => Arbitrary (SumSpec fn a b) where
+  arbitrary = SumSpec <$> arbitrary <*> arbitrary
+  shrink (SumSpec a b) = uncurry SumSpec <$> shrink (a, b)
+
 instance (HasSpec fn a, HasSpec fn b) => Semigroup (SumSpec fn a b) where
   SumSpec sa sb <> SumSpec sa' sb' = SumSpec (sa <> sa') (sb <> sb')
 

diff --git a/libs/constrained-generators/src/Constrained/Properties.hs b/libs/constrained-generators/src/Constrained/Properties.hs
@@ -13,6 +13,7 @@
 module Constrained.Properties where
 
 import Data.Map (Map)
+import Data.Set (Set)
 import Data.Typeable
 import Constrained.Base
 import Constrained.List
@@ -151,6 +152,9 @@ instance (fn ~ BaseFn) => QC.Arbitrary (TestableFn fn) where
   arbitrary = QC.elements [ TestableFn $ addFn @fn @Int
                           , TestableFn $ negateFn @fn @Int
                           , TestableFn $ sizeOfFn @fn @(Map Int Int)
+                          , TestableFn $ memberFn @fn @(Set Int)
+                          , TestableFn $ notFn @fn
+                          , TestableFn $ disjointFn @fn @(Set Int)
                           ]
 
 prop_mapSpec ::