WIP

cardano-foundation · May 3, 2024 · 94162fa · 94162fa
1 parent 25930e5
commit 94162fa
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/lib/balance-tx/test/spec/Internal/Cardano/Write/Tx/BalanceSpec.hs b/lib/balance-tx/test/spec/Internal/Cardano/Write/Tx/BalanceSpec.hs
@@ -2213,25 +2213,22 @@ instance Arbitrary (MixedSign Value) where
 
 instance forall era. IsRecentEra era => Arbitrary (PartialTx era) where
     arbitrary =
-        txWithUTxOToPartialTx <$>
-        TxWithUTxO.generate genTxForBalancing genTxIn genTxOut
-    shrink = shrinkMapBy
-        txWithUTxOToPartialTx
-        txWithUTxOFromPartialTx
-        TxWithUTxO.shrink
+        mkPartialTx <$> TxWithUTxO.generate genTxForBalancing genTxIn genTxOut
+    shrink =
+        shrinkMapBy mkPartialTx unPartialTx TxWithUTxO.shrink
 
 genTxIn :: Gen TxIn
 genTxIn = fromWalletTxIn <$> W.genTxIn
 
-txWithUTxOToPartialTx :: IsRecentEra era => TxWithUTxO era -> PartialTx era
-txWithUTxOToPartialTx (TxWithUTxO tx extraUTxO) =
+mkPartialTx :: IsRecentEra era => TxWithUTxO era -> PartialTx era
+mkPartialTx (TxWithUTxO tx extraUTxO) =
     PartialTx {tx, extraUTxO, redeemers, timelockKeyWitnessCounts}
   where
     redeemers = []
     timelockKeyWitnessCounts = mempty
 
-txWithUTxOFromPartialTx :: IsRecentEra era => PartialTx era -> TxWithUTxO era
-txWithUTxOFromPartialTx PartialTx {tx, extraUTxO} =
+unPartialTx :: IsRecentEra era => PartialTx era -> TxWithUTxO era
+unPartialTx PartialTx {tx, extraUTxO} =
     TxWithUTxO.constructFiltered tx extraUTxO
 
 instance Arbitrary StdGenSeed  where