@@ -249,27 +249,21 @@ def test_multi_multi():
249
249
pubkeys = [privtopub (k ) for k in privkeys ]
250
250
251
251
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 )
255
255
self .agg_pubkeys = [
256
256
aggregate_pubkeys ([pubkeys [i ] for i in msg_1_validators ]),
257
257
aggregate_pubkeys ([pubkeys [i ] for i in msg_2_validators ]),
258
258
]
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 ]
262
262
self .sig = aggregate_signatures ([
263
263
aggregate_signatures (msg_1_sigs ),
264
264
aggregate_signatures (msg_2_sigs ),
265
265
])
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
273
267
assert verify_multiple_multiple (
274
268
signatures = [att .sig for att in atts ],
275
269
pubkeys_and_messages = [[att .agg_pubkeys , att .msgs ] for att in atts ],
0 commit comments