diff options
-rw-r--r-- | test/functional/test_framework/address.py | 4 | ||||
-rw-r--r-- | test/functional/test_framework/segwit_addr.py | 10 |
2 files changed, 7 insertions, 7 deletions
diff --git a/test/functional/test_framework/address.py b/test/functional/test_framework/address.py index 9506b63f82..536647534a 100644 --- a/test/functional/test_framework/address.py +++ b/test/functional/test_framework/address.py @@ -8,9 +8,9 @@ import enum import unittest from .script import hash256, hash160, sha256, CScript, OP_0 +from .segwit_addr import encode_segwit_address from .util import hex_str_to_bytes -from . import segwit_addr from test_framework.util import assert_equal @@ -100,7 +100,7 @@ def program_to_witness(version, program, main = False): assert 0 <= version <= 16 assert 2 <= len(program) <= 40 assert version > 0 or len(program) in [20, 32] - return segwit_addr.encode("bc" if main else "bcrt", version, program) + return encode_segwit_address("bc" if main else "bcrt", version, program) def script_to_p2wsh(script, main = False): script = check_script(script) diff --git a/test/functional/test_framework/segwit_addr.py b/test/functional/test_framework/segwit_addr.py index 10d20eeda7..00c0d8a919 100644 --- a/test/functional/test_framework/segwit_addr.py +++ b/test/functional/test_framework/segwit_addr.py @@ -84,7 +84,7 @@ def convertbits(data, frombits, tobits, pad=True): return ret -def decode(hrp, addr): +def decode_segwit_address(hrp, addr): """Decode a segwit address.""" hrpgot, data = bech32_decode(addr) if hrpgot != hrp: @@ -99,10 +99,10 @@ def decode(hrp, addr): return (data[0], decoded) -def encode(hrp, witver, witprog): +def encode_segwit_address(hrp, witver, witprog): """Encode a segwit address.""" ret = bech32_encode(hrp, [witver] + convertbits(witprog, 8, 5)) - if decode(hrp, ret) == (None, None): + if decode_segwit_address(hrp, ret) == (None, None): return None return ret @@ -111,8 +111,8 @@ class TestFrameworkScript(unittest.TestCase): def test_python_bech32(addr): hrp = addr[:4] self.assertEqual(hrp, "bcrt") - (witver, witprog) = decode(hrp, addr) - self.assertEqual(encode(hrp, witver, witprog), addr) + (witver, witprog) = decode_segwit_address(hrp, addr) + self.assertEqual(encode_segwit_address(hrp, witver, witprog), addr) # P2WPKH test_python_bech32('bcrt1qthmht0k2qnh3wy7336z05lu2km7emzfpm3wg46') |