highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-12-08 19:53:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

test: refactor: use random.sample for choosing random keys in wallet_taproot.py

Browse Source
This commit is contained in:
Sebastian Falbesoner
2022-03-07 18:40:17 +01:00
parent c9ed9927bb
commit 31846b006d
1 changed files with 1 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/functional/wallet_taproot.py
View File

@@ -207,19 +207,6 @@ class WalletTaprootTest(BitcoinTestFramework):
def init_wallet(self, *, node): def init_wallet(self, *, node):
pass pass
@staticmethod
def rand_keys(n):
ret = []
idxes = set()
for _ in range(n):
while True:
i = random.randrange(len(KEYS))
if not i in idxes:
break
idxes.add(i)
ret.append(KEYS[i])
return ret
@staticmethod @staticmethod
def make_desc(pattern, privmap, keys, pub_only = False): def make_desc(pattern, privmap, keys, pub_only = False):
pat = pattern.replace("$H", H_POINT) pat = pattern.replace("$H", H_POINT)
@@ -332,7 +319,7 @@ class WalletTaprootTest(BitcoinTestFramework):
def do_test(self, comment, pattern, privmap, treefn): def do_test(self, comment, pattern, privmap, treefn):
nkeys = len(privmap) nkeys = len(privmap)
keys = self.rand_keys(nkeys * 4) keys = random.sample(KEYS, nkeys * 4)
self.do_test_addr(comment, pattern, privmap, treefn, keys[0:nkeys]) self.do_test_addr(comment, pattern, privmap, treefn, keys[0:nkeys])
self.do_test_sendtoaddress(comment, pattern, privmap, treefn, keys[0:nkeys], keys[nkeys:2*nkeys]) self.do_test_sendtoaddress(comment, pattern, privmap, treefn, keys[0:nkeys], keys[nkeys:2*nkeys])
self.do_test_psbt(comment, pattern, privmap, treefn, keys[2*nkeys:3*nkeys], keys[3*nkeys:4*nkeys]) self.do_test_psbt(comment, pattern, privmap, treefn, keys[2*nkeys:3*nkeys], keys[3*nkeys:4*nkeys])