Property based testing for potential inputs

Author: cffls

test/pycardano/test_txbuilder.py

@@ -1,3 +1,4 @@
+import copy
 from dataclasses import replace
 from test.pycardano.test_key import SK
 from test.pycardano.util import chain_context
@@ -129,30 +130,24 @@ def test_tx_builder_with_potential_inputs(chain_context):
 
     utxos = chain_context.utxos(sender)
 
-    tx_builder.potential_inputs.append(utxos[1])
+    tx_builder.potential_inputs.extend(utxos)
 
-    tx_builder.add_output(TransactionOutput.from_primitive([sender, 500000]))
+    for i in range(20):
+        utxo = copy.deepcopy(utxos[0])
+        utxo.input.index = i + 100
+        tx_builder.potential_inputs.append(utxo)
 
-    tx_body = tx_builder.build(change_address=sender_address)
+    assert len(tx_builder.potential_inputs) > 1
 
-    expected = {
-        0: [[b"22222222222222222222222222222222", 1]],
-        1: [
-            # First output
-            [sender_address.to_primitive(), 500000],
-            # Second output as change
-            [
-                sender_address.to_primitive(),
-                [
-                    5332431,
-                    {b"1111111111111111111111111111": {b"Token1": 1, b"Token2": 2}},
-                ],
-            ],
-        ],
-        2: 167569,
-    }
+    tx_builder.add_output(
+        TransactionOutput.from_primitive(
+            [sender, [5000000, {b"1111111111111111111111111111": {b"Token1": 1}}]]
+        )
+    )
 
-    assert expected == tx_body.to_primitive()
+    tx_body = tx_builder.build(change_address=sender_address)
+
+    assert len(tx_body.inputs) < len(tx_builder.potential_inputs)
 
 
 def test_tx_builder_multi_asset(chain_context):