@@ -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