Add set_wif to test frameworks's ECKey

test/functional/test_framework/key.py

@@ -8,6 +8,8 @@
 anything but tests."""
 import random
 
+from .address import base58_to_byte
+
 def modinv(a, n):
     """Compute the modular inverse of a modulo n
 
@@ -337,6 +339,13 @@ def set(self, secret, compressed):
             self.secret = secret
             self.compressed = compressed
 
+    def set_wif(self, wif):
+        (b, v) = base58_to_byte(wif)
+        if len(b) != 32 and len(b) != 33:
+            raise ValueError("invalid WIF: unexpected length {}".format(len(b)))
+        compressed = len(b) == 33
+        self.set(b[0:32], compressed)
+
     def generate(self, compressed=True):
         """Generate a random private key (compressed or uncompressed)."""
         self.set(random.randrange(1, SECP256K1_ORDER).to_bytes(32, 'big'), compressed)