Remove obvious TokenBundle.Gen weeds

cardano-foundation · Jul 20, 2021 · cfce60c · cfce60c
1 parent c3be768
commit cfce60c
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 60 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
@@ -8,9 +8,7 @@ module Cardano.Wallet.Primitive.Types.TokenMap.Gen
     , genAssetIdSmallRange
     , genTokenMapSized
     , genTokenMapSmallRange
-    , shrinkAssetIdSized
     , shrinkAssetIdSmallRange
-    , shrinkTokenMapSized
     , shrinkTokenMapSmallRange
     , AssetIdF (..)
     ) where
@@ -26,19 +24,13 @@ import Cardano.Wallet.Primitive.Types.TokenPolicy.Gen
     , genTokenPolicyIdLargeRange
     , genTokenPolicyIdSized
     , genTokenPolicyIdSmallRange
-    , shrinkTokenNameSized
     , shrinkTokenNameSmallRange
-    , shrinkTokenPolicyIdSized
     , shrinkTokenPolicyIdSmallRange
     , tokenNamesMediumRange
     , tokenPolicies
     )
 import Cardano.Wallet.Primitive.Types.TokenQuantity.Gen
-    ( genTokenQuantitySized
-    , genTokenQuantitySmall
-    , shrinkTokenQuantitySized
-    , shrinkTokenQuantitySmall
-    )
+    ( genTokenQuantitySized, genTokenQuantitySmall, shrinkTokenQuantitySmall )
 import Control.Monad
     ( replicateM )
 import Data.List
@@ -86,11 +78,6 @@ genAssetIdSized = sized $ \size -> do
         <$> resize sizeSquareRoot genTokenPolicyIdSized
         <*> resize sizeSquareRoot genTokenNameSized
 
-shrinkAssetIdSized :: AssetId -> [AssetId]
-shrinkAssetIdSized (AssetId p t) = uncurry AssetId <$> shrinkInterleaved
-    (p, shrinkTokenPolicyIdSized)
-    (t, shrinkTokenNameSized)
-
 --------------------------------------------------------------------------------
 -- Asset identifiers chosen from a small range (to allow collisions)
 --------------------------------------------------------------------------------
@@ -128,16 +115,6 @@ genTokenMapSized = sized $ \size -> do
         <$> genAssetIdSized
         <*> genTokenQuantitySized
 
-shrinkTokenMapSized :: TokenMap -> [TokenMap]
-shrinkTokenMapSized
-    = fmap TokenMap.fromFlatList
-    . shrinkList shrinkAssetQuantity
-    . TokenMap.toFlatList
-  where
-    shrinkAssetQuantity (a, q) = shrinkInterleaved
-        (a, shrinkAssetIdSized)
-        (q, shrinkTokenQuantitySized)
-
 --------------------------------------------------------------------------------
 -- Token maps with assets and quantities chosen from small ranges
 --------------------------------------------------------------------------------

diff --git a/lib/core/src/Cardano/Wallet/Primitive/Types/TokenPolicy/Gen.hs b/lib/core/src/Cardano/Wallet/Primitive/Types/TokenPolicy/Gen.hs
@@ -7,10 +7,7 @@ module Cardano.Wallet.Primitive.Types.TokenPolicy.Gen
     , genTokenPolicyIdLargeRange
     , genTokenPolicyIdSmallRange
     , mkTokenPolicyId
-    , shrinkTokenNameSized
-    , shrinkTokenNameMediumRange
     , shrinkTokenNameSmallRange
-    , shrinkTokenPolicyIdSized
     , shrinkTokenPolicyIdSmallRange
     , tokenNamesMediumRange
     , tokenNamesSmallRange
@@ -42,13 +39,6 @@ genTokenNameSized :: Gen TokenName
 genTokenNameSized = sized $ \size ->
     elements $ UnsafeTokenName . B8.snoc "Token" <$> take size ['A' ..]
 
-shrinkTokenNameSized :: TokenName -> [TokenName]
-shrinkTokenNameSized x
-    | x == simplest = []
-    | otherwise = [simplest]
-  where
-    simplest = UnsafeTokenName "TokenA"
-
 --------------------------------------------------------------------------------
 -- Token names chosen from a small range (to allow collisions)
 --------------------------------------------------------------------------------
@@ -74,13 +64,6 @@ tokenNamesSmallRange = UnsafeTokenName . B8.snoc "Token" <$> ['A' .. 'D']
 genTokenNameMediumRange :: Gen TokenName
 genTokenNameMediumRange = elements tokenNamesMediumRange
 
-shrinkTokenNameMediumRange :: TokenName -> [TokenName]
-shrinkTokenNameMediumRange x
-    | x == simplest = []
-    | otherwise = [simplest]
-  where
-    simplest = head tokenNamesMediumRange
-
 tokenNamesMediumRange :: [TokenName]
 tokenNamesMediumRange = UnsafeTokenName . B8.snoc "Token" <$> ['A' .. 'Z']
 
@@ -100,13 +83,6 @@ genTokenPolicyIdSized :: Gen TokenPolicyId
 genTokenPolicyIdSized = sized $ \size ->
     elements $ mkTokenPolicyId <$> take size mkTokenPolicyIdValidChars
 
-shrinkTokenPolicyIdSized :: TokenPolicyId -> [TokenPolicyId]
-shrinkTokenPolicyIdSized x
-    | x == simplest = []
-    | otherwise = [simplest]
-  where
-    simplest = mkTokenPolicyId 'A'
-
 --------------------------------------------------------------------------------
 -- Token policy identifiers chosen from a small range (to allow collisions)
 --------------------------------------------------------------------------------

diff --git a/lib/core/src/Cardano/Wallet/Primitive/Types/TokenQuantity/Gen.hs b/lib/core/src/Cardano/Wallet/Primitive/Types/TokenQuantity/Gen.hs
@@ -5,11 +5,8 @@ module Cardano.Wallet.Primitive.Types.TokenQuantity.Gen
     , genTokenQuantityLarge
     , genTokenQuantityMassive
     , genTokenQuantityMixed
-    , shrinkTokenQuantitySized
     , shrinkTokenQuantitySmall
     , shrinkTokenQuantitySmallPositive
-    , shrinkTokenQuantityLarge
-    , shrinkTokenQuantityMassive
     , shrinkTokenQuantityMixed
     , tokenQuantitySmall
     , tokenQuantityLarge
@@ -35,9 +32,6 @@ genTokenQuantitySized :: Gen TokenQuantity
 genTokenQuantitySized = sized $ \n ->
     quantityFromInt <$> choose (0, n)
 
-shrinkTokenQuantitySized :: TokenQuantity -> [TokenQuantity]
-shrinkTokenQuantitySized = shrinkTokenQuantity
-
 --------------------------------------------------------------------------------
 -- Small token quantities
 --------------------------------------------------------------------------------
@@ -73,9 +67,6 @@ genTokenQuantityLarge = quantityFromInteger <$> choose
     , quantityToInteger tokenQuantityLarge
     )
 
-shrinkTokenQuantityLarge :: TokenQuantity -> [TokenQuantity]
-shrinkTokenQuantityLarge = shrinkTokenQuantity
-
 --------------------------------------------------------------------------------
 -- Massive token quantities
 --------------------------------------------------------------------------------
@@ -86,9 +77,6 @@ genTokenQuantityMassive = quantityFromInteger <$> choose
     , quantityToInteger tokenQuantityMassive
     )
 
-shrinkTokenQuantityMassive :: TokenQuantity -> [TokenQuantity]
-shrinkTokenQuantityMassive = shrinkTokenQuantity
-
 --------------------------------------------------------------------------------
 -- Mixed token quantities (both small and large)
 --------------------------------------------------------------------------------