status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

mock attestation refactor

This commit is contained in:
Carl Beekhuizen 2019-06-29 15:48:39 +02:00
parent 0680d8cc53
commit 4ed7af7bac
No known key found for this signature in database
GPG Key ID: D05CA176D0020646
1 changed files with 25 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
test_libs/pyspec/eth2spec/test/phase_0/epoch_processing/test_process_justification_and_finalization.py
View File

@ -1,4 +1,3 @@
import math
from eth2spec.test.context import spec_state_test, with_all_phases from eth2spec.test.context import spec_state_test, with_all_phases
from eth2spec.test.phase_0.epoch_processing.run_epoch_process_base import ( from eth2spec.test.phase_0.epoch_processing.run_epoch_process_base import (
run_epoch_processing_with run_epoch_processing_with
@ -39,18 +38,26 @@ def add_mock_attestations(spec, state, epoch, source, target, sufficient_support
else: else:
raise Exception(f"cannot include attestations in epoch ${epoch} from epoch ${current_epoch}") raise Exception(f"cannot include attestations in epoch ${epoch} from epoch ${current_epoch}")
committee_count = spec.get_epoch_committee_count(state, epoch) total_balance = spec.get_total_active_balance(state)
indices = spec.get_active_validator_indices(state, epoch) remaining_balance = total_balance * 2 // 3
epoch_start_shard = spec.get_epoch_start_shard(state, epoch)
epoch_start_slot = spec.get_epoch_start_slot(epoch) epoch_start_slot = spec.get_epoch_start_slot(epoch)
for slot in range(epoch_start_slot, epoch_start_slot + spec.SLOTS_PER_EPOCH): for slot in range(epoch_start_slot, epoch_start_slot + spec.SLOTS_PER_EPOCH):
for shard in get_shards_for_slot(spec, state, slot): for shard in get_shards_for_slot(spec, state, slot):
size = get_committee_size(spec, epoch_start_shard, shard, committee_count, indices) committee = spec.get_crosslink_committee(state, spec.slot_to_epoch(slot), shard)
# Create a bitfield filled with the given count per attestation, # Create a bitfield filled with the given count per attestation,
# exactly on the right-most part of the committee field. # exactly on the right-most part of the committee field.
attesting_count = math.ceil(size * 2 /3)
attesting_count = attesting_count if sufficient_support else attesting_count - 1 aggregation_bits = [0] * len(committee)
aggregation_bits = [i < attesting_count for i in range(size)] for v in range(len(committee) * 2 // 3 + 1):
if remaining_balance > 0:
remaining_balance -= state.validators[v].effective_balance
aggregation_bits[v] = 1
elif not sufficient_support:
aggregation_bits[v - 1] = 0
break
else:
break
attestations.append(spec.PendingAttestation( attestations.append(spec.PendingAttestation(
aggregation_bits=aggregation_bits, aggregation_bits=aggregation_bits,