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

rewards testing now with cleaner attestation signing

This commit is contained in:
protolambda 2019-09-24 13:56:29 +09:00
parent a6e543fd37
commit 525d733033
No known key found for this signature in database
GPG Key ID: EC89FDBB2B4C7623
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test_libs/pyspec/eth2spec/test/phase_0/epoch_processing/test_process_rewards_and_penalties.py
View File

@ -7,7 +7,6 @@ from eth2spec.test.helpers.state import (
) )
from eth2spec.test.helpers.attestations import ( from eth2spec.test.helpers.attestations import (
add_attestations_to_state, add_attestations_to_state,
fill_aggregate_attestation,
get_valid_attestation, get_valid_attestation,
) )
from eth2spec.test.phase_0.epoch_processing.run_epoch_process_base import run_epoch_processing_with from eth2spec.test.phase_0.epoch_processing.run_epoch_process_base import run_epoch_processing_with
@ -37,8 +36,7 @@ def test_genesis_epoch_full_attestations_no_rewards(spec, state):
for slot in range(spec.SLOTS_PER_EPOCH - 1): for slot in range(spec.SLOTS_PER_EPOCH - 1):
# create an attestation for each slot # create an attestation for each slot
if slot < spec.SLOTS_PER_EPOCH: if slot < spec.SLOTS_PER_EPOCH:
attestation = get_valid_attestation(spec, state) attestation = get_valid_attestation(spec, state, signed=True)
fill_aggregate_attestation(spec, state, attestation, signed=True)
attestations.append(attestation) attestations.append(attestation)
# fill each created slot in state after inclusion delay # fill each created slot in state after inclusion delay
if slot - spec.MIN_ATTESTATION_INCLUSION_DELAY >= 0: if slot - spec.MIN_ATTESTATION_INCLUSION_DELAY >= 0:
@ -64,8 +62,7 @@ def test_full_attestations(spec, state):
for slot in range(spec.SLOTS_PER_EPOCH + spec.MIN_ATTESTATION_INCLUSION_DELAY): for slot in range(spec.SLOTS_PER_EPOCH + spec.MIN_ATTESTATION_INCLUSION_DELAY):
# create an attestation for each slot in epoch # create an attestation for each slot in epoch
if slot < spec.SLOTS_PER_EPOCH: if slot < spec.SLOTS_PER_EPOCH:
attestation = get_valid_attestation(spec, state) attestation = get_valid_attestation(spec, state, signed=True)
fill_aggregate_attestation(spec, state, attestation, signed=True)
attestations.append(attestation) attestations.append(attestation)
# fill each created slot in state after inclusion delay # fill each created slot in state after inclusion delay
if slot - spec.MIN_ATTESTATION_INCLUSION_DELAY >= 0: if slot - spec.MIN_ATTESTATION_INCLUSION_DELAY >= 0:
@ -112,8 +109,7 @@ def test_duplicate_attestation(spec, state):
This test addresses this issue found at Interop This test addresses this issue found at Interop
https://github.com/djrtwo/interop-test-cases/tree/master/tests/prysm_16_duplicate_attestation_rewards https://github.com/djrtwo/interop-test-cases/tree/master/tests/prysm_16_duplicate_attestation_rewards
""" """
attestation = get_valid_attestation(spec, state) attestation = get_valid_attestation(spec, state, signed=True)
fill_aggregate_attestation(spec, state, attestation, signed=True)
indexed_attestation = spec.get_indexed_attestation(state, attestation) indexed_attestation = spec.get_indexed_attestation(state, attestation)
participants = indexed_attestation.custody_bit_0_indices + indexed_attestation.custody_bit_1_indices participants = indexed_attestation.custody_bit_0_indices + indexed_attestation.custody_bit_1_indices
@ -131,7 +127,8 @@ def test_duplicate_attestation(spec, state):
next_epoch(spec, dup_state) next_epoch(spec, dup_state)
# Run non-duplicate inclusion rewards for comparision. Do not yield test vectors # Run non-duplicate inclusion rewards for comparision. Do not yield test vectors
pre, post = run_process_rewards_and_penalties(spec, single_state) for _ in run_process_rewards_and_penalties(spec, single_state):
pass
# Output duplicate inclusion to test vectors # Output duplicate inclusion to test vectors
yield from run_process_rewards_and_penalties(spec, dup_state) yield from run_process_rewards_and_penalties(spec, dup_state)