Re-add arbitrary pparam in shrink property

cardano-foundation · Sep 13, 2023 · 9f1c9ad · 9f1c9ad
1 parent 4aa8a83
commit 9f1c9ad
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/wallet/test/unit/Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs b/lib/wallet/test/unit/Cardano/Wallet/Write/Tx/Balance/TokenBundleSizeSpec.hs
@@ -108,16 +108,17 @@ prop_assessTokenBundleSize_enlarge b1' b2' pp =
 prop_assessTokenBundleSize_shrink
     :: Blind (VariableSize1024 TokenBundle)
     -> Blind (VariableSize16 TokenBundle)
+    -> PParamsInRecentEra
     -> Property
-prop_assessTokenBundleSize_shrink b1' b2' =
+prop_assessTokenBundleSize_shrink b1' b2' pp =
     assess b1 == TokenBundleSizeWithinLimit ==> conjoin
         [ assess (b1 `TokenBundle.difference` b2)
             === TokenBundleSizeWithinLimit
         , assess (b1 `TokenBundle.setCoin` txOutMinCoin)
             === TokenBundleSizeWithinLimit
         ]
   where
-    assess = assessTokenBundleSize babbageTokenBundleSizeAssessor
+    assess = assessTokenBundleSize $ mkAssessorFromPParamsInRecentEra pp
     b1 = unVariableSize1024 $ getBlind b1'
     b2 = unVariableSize16 $ getBlind b2'