status-im/eth2.0-specs
2
0
Fork 0
mirror of https://github.com/status-im/eth2.0-specs.git synced 2025-01-19 23:19:28 +00:00
Code Issues Projects Releases Wiki Activity

update test to work for both minimal and mainnet config

Browse Source
This commit is contained in:
Alex Stokes 2021-04-08 18:25:58 -07:00
parent e4e65295b6
commit 66905f4fa6
No known key found for this signature in database
GPG Key ID: 99B3D88FD6C55A69
1 changed files with 24 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tests/core/pyspec/eth2spec/test/altair/unittests/validator/test_validator.py
View File

@ -1,4 +1,5 @@
import random
from collections import defaultdict
from eth2spec.utils.ssz.ssz_typing import Bitvector
from eth2spec.test.helpers.block import build_empty_block
from eth2spec.test.helpers.keys import pubkey_to_privkey
@ -135,24 +136,31 @@ def test_process_sync_committee_contributions(phases, spec, state):
spec.process_block(state, block)
def _validator_index_for_pubkey(state, pubkey):
return list(map(lambda v: v.pubkey, state.validators)).index(pubkey)
def _subnet_for_sync_committee_index(spec, i):
return i // (spec.SYNC_COMMITTEE_SIZE // spec.SYNC_COMMITTEE_SUBNET_COUNT)
@with_all_phases_except([PHASE0])
@with_state
def test_compute_subnets_for_sync_committee(state, spec, phases):
k = max(spec.SYNC_COMMITTEE_SIZE // 4, 1)
some_sync_committee_members = rng.sample(
list(
(i, pubkey) for i, pubkey in enumerate(state.current_sync_committee.pubkeys)
),
k,
some_sync_committee_members = list(
(
_subnet_for_sync_committee_index(spec, i),
pubkey,
)
for i, pubkey in enumerate(state.current_sync_committee.pubkeys)
)
validator_indices = [
list(map(lambda v: v.pubkey, state.validators)).index(pubkey)
for pubkey in map(lambda t: t[1], some_sync_committee_members)
]
expected_subnets = [
index // (spec.SYNC_COMMITTEE_SIZE // spec.SYNC_COMMITTEE_SUBNET_COUNT)
for index in map(lambda t: t[0], some_sync_committee_members)
]
for validator_index, expected_subnet in zip(validator_indices, expected_subnets):
expected_subnets_by_pubkey = defaultdict(list)
for (subnet, pubkey) in some_sync_committee_members:
expected_subnets_by_pubkey[pubkey].append(subnet)
for _, pubkey in some_sync_committee_members:
validator_index = _validator_index_for_pubkey(state, pubkey)
subnets = spec.compute_subnets_for_sync_committee(state, validator_index)
assert subnets == [expected_subnet]
expected_subnets = expected_subnets_by_pubkey[pubkey]
assert subnets == expected_subnets