diff --git a/test/functional/feature_vaults.py b/test/functional/feature_vaults.py index 11472a6e0b4ea..8ac4e8238a322 100755 --- a/test/functional/feature_vaults.py +++ b/test/functional/feature_vaults.py @@ -348,6 +348,7 @@ def test_batch_unvault( # Construct the final unvault target. unvault_total_sats = sum(v.total_amount_sats for v in vaults) # type: ignore + assert unvault_total_sats is not None assert unvault_total_sats > 0 target_key = key.ECKey(secret=(1).to_bytes(32, "big")) @@ -413,8 +414,11 @@ def test_revault( revault_amount = COIN # Construct the final unvault target; save a coin to peel off in the revault. - unvault_total_sats = sum( - v.total_amount_sats for v in vaults) - revault_amount # type: ignore + unvault_total_sats = 0 + for v in vaults: + assert v.total_amount_sats is not None + unvault_total_sats += v.total_amount_sats + unvault_total_sats -= revault_amount assert unvault_total_sats > 0 target_key = key.ECKey(secret=(1).to_bytes(32, "big")) @@ -1167,6 +1171,7 @@ def __init__( self.trigger_spk = self.trigger_tr.scriptPubKey self.amount = sum(v.total_amount_sats for v in self.compat_vaults) # type: ignore + assert self.amount is not None if BadTriggerAmountLow in self.bad_behavior: self.amount -= 1