Skip to content

Commit fc65be3

Browse files
PR feedback 2, remove randomness
1 parent 59c319d commit fc65be3

File tree

1 file changed

+7
-13
lines changed

1 file changed

+7
-13
lines changed

tests/test_bls.py

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -249,27 +249,21 @@ def test_multi_multi():
249249
pubkeys = [privtopub(k) for k in privkeys]
250250

251251
class Attestation:
252-
def __init__(self, msg_1, msg_2):
253-
msg_1_validators = sample(validator_indices, 3)
254-
msg_2_validators = sample(validator_indices, 3)
252+
def __init__(self):
253+
msg_1_validators = (1, 2, 3, 4)
254+
msg_2_validators = (4, 5, 6, 7)
255255
self.agg_pubkeys = [
256256
aggregate_pubkeys([pubkeys[i] for i in msg_1_validators]),
257257
aggregate_pubkeys([pubkeys[i] for i in msg_2_validators]),
258258
]
259-
self.msgs = [msg_1, msg_2]
260-
msg_1_sigs = [sign(msg_1, privkeys[i], domain) for i in msg_1_validators]
261-
msg_2_sigs = [sign(msg_2, privkeys[i], domain) for i in msg_2_validators]
259+
self.msgs = (b'\x12' * 32, b'\x34' * 32)
260+
msg_1_sigs = [sign(self.msgs[0], privkeys[i], domain) for i in msg_1_validators]
261+
msg_2_sigs = [sign(self.msgs[1], privkeys[i], domain) for i in msg_2_validators]
262262
self.sig = aggregate_signatures([
263263
aggregate_signatures(msg_1_sigs),
264264
aggregate_signatures(msg_2_sigs),
265265
])
266-
267-
msgs = (
268-
(b'\x12' * 32, b'\x34' * 32),
269-
(b'\x56' * 32, b'\x78' * 32),
270-
(b'\x90' * 32, b'\xab' * 32),
271-
)
272-
atts = [Attestation(msg_1, msg_2) for msg_1, msg_2 in msgs]
266+
atts = (Attestation(),) * 3
273267
assert verify_multiple_multiple(
274268
signatures=[att.sig for att in atts],
275269
pubkeys_and_messages=[[att.agg_pubkeys, att.msgs] for att in atts],

0 commit comments

Comments
 (0)