Increase the chance of generating empty or singleton token maps.

cardano-foundation · Jan 13, 2021 · 8280023 · 8280023
1 parent b946655
commit 8280023
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/core/src/Cardano/Wallet/Primitive/Types/TokenMap/Gen.hs b/lib/core/src/Cardano/Wallet/Primitive/Types/TokenMap/Gen.hs
@@ -20,7 +20,7 @@ import Cardano.Wallet.Primitive.Types.TokenQuantity.Gen
 import Control.Monad
     ( replicateM )
 import Test.QuickCheck
-    ( Gen, choose, shrinkList )
+    ( Gen, choose, oneof, shrinkList )
 import Test.QuickCheck.Extra
     ( shrinkInterleaved )
 
@@ -46,7 +46,11 @@ shrinkAssetIdSmallRange (AssetId p t) = uncurry AssetId <$> shrinkInterleaved
 
 genTokenMapSmallRange :: Gen TokenMap
 genTokenMapSmallRange = do
-    assetCount <- choose (0, 16)
+    assetCount <- oneof
+        [ pure 0
+        , pure 1
+        , choose (2, 16)
+        ]
     TokenMap.fromFlatList <$> replicateM assetCount genAssetQuantity
   where
     genAssetQuantity = (,)