Skip to content

Commit 31846b0

Browse files
committed
test: refactor: use random.sample for choosing random keys in wallet_taproot.py
1 parent c9ed992 commit 31846b0

File tree

1 file changed

+1
-14
lines changed

1 file changed

+1
-14
lines changed

test/functional/wallet_taproot.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -207,19 +207,6 @@ def setup_network(self):
207207
def init_wallet(self, *, node):
208208
pass
209209

210-
@staticmethod
211-
def rand_keys(n):
212-
ret = []
213-
idxes = set()
214-
for _ in range(n):
215-
while True:
216-
i = random.randrange(len(KEYS))
217-
if not i in idxes:
218-
break
219-
idxes.add(i)
220-
ret.append(KEYS[i])
221-
return ret
222-
223210
@staticmethod
224211
def make_desc(pattern, privmap, keys, pub_only = False):
225212
pat = pattern.replace("$H", H_POINT)
@@ -332,7 +319,7 @@ def do_test_psbt(self, comment, pattern, privmap, treefn, keys_pay, keys_change)
332319

333320
def do_test(self, comment, pattern, privmap, treefn):
334321
nkeys = len(privmap)
335-
keys = self.rand_keys(nkeys * 4)
322+
keys = random.sample(KEYS, nkeys * 4)
336323
self.do_test_addr(comment, pattern, privmap, treefn, keys[0:nkeys])
337324
self.do_test_sendtoaddress(comment, pattern, privmap, treefn, keys[0:nkeys], keys[nkeys:2*nkeys])
338325
self.do_test_psbt(comment, pattern, privmap, treefn, keys[2*nkeys:3*nkeys], keys[3*nkeys:4*nkeys])

0 commit comments

Comments
 (0)