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

Merge pull request #2503 from ethereum/flag-updates-tests 

add more test_process_participation_flag_updates tests
This commit is contained in:
Danny Ryan 2021-07-02 09:25:43 -06:00 committed by GitHub
parent 060ba633eb 927b26636e
commit cdc3e6756f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 48 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
tests/core/pyspec/eth2spec/test/altair/epoch_processing/test_process_participation_flag_updates.py
View File

@ -12,6 +12,13 @@ from eth2spec.test.helpers.state import next_epoch_via_block
from eth2spec.test.helpers.epoch_processing import run_epoch_processing_with from eth2spec.test.helpers.epoch_processing import run_epoch_processing_with
def get_full_flags(spec):
full_flags = spec.ParticipationFlags(0)
for flag_index in range(len(spec.PARTICIPATION_FLAG_WEIGHTS)):
full_flags = spec.add_flag(full_flags, flag_index)
return full_flags
def run_process_participation_flag_updates(spec, state): def run_process_participation_flag_updates(spec, state):
old = state.current_epoch_participation.copy() old = state.current_epoch_participation.copy()
yield from run_epoch_processing_with(spec, state, 'process_participation_flag_updates') yield from run_epoch_processing_with(spec, state, 'process_participation_flag_updates')
@ -33,12 +40,30 @@ def test_all_zeroed(spec, state):
def test_filled(spec, state): def test_filled(spec, state):
next_epoch_via_block(spec, state) next_epoch_via_block(spec, state)
full_flags = spec.ParticipationFlags(0) state.previous_epoch_participation = [get_full_flags(spec)] * len(state.validators)
for flag_index in range(len(spec.PARTICIPATION_FLAG_WEIGHTS)): state.current_epoch_participation = [get_full_flags(spec)] * len(state.validators)
full_flags = spec.add_flag(full_flags, flag_index)
state.previous_epoch_participation = [full_flags] * len(state.validators) yield from run_process_participation_flag_updates(spec, state)
state.current_epoch_participation = [full_flags] * len(state.validators)
@with_altair_and_later
@spec_state_test
def test_previous_filled(spec, state):
next_epoch_via_block(spec, state)
state.previous_epoch_participation = [get_full_flags(spec)] * len(state.validators)
state.current_epoch_participation = [0] * len(state.validators)
yield from run_process_participation_flag_updates(spec, state)
@with_altair_and_later
@spec_state_test
def test_current_filled(spec, state):
next_epoch_via_block(spec, state)
state.previous_epoch_participation = [0] * len(state.validators)
state.current_epoch_participation = [get_full_flags(spec)] * len(state.validators)
yield from run_process_participation_flag_updates(spec, state) yield from run_process_participation_flag_updates(spec, state)
@ -55,9 +80,25 @@ def random_flags(spec, state, seed: int, previous=True, current=True):
@with_altair_and_later @with_altair_and_later
@spec_state_test @spec_state_test
def test_random(spec, state): def test_random_0(spec, state):
next_epoch_via_block(spec, state) next_epoch_via_block(spec, state)
random_flags(spec, state, 10) random_flags(spec, state, 100)
yield from run_process_participation_flag_updates(spec, state)
@with_altair_and_later
@spec_state_test
def test_random_1(spec, state):
next_epoch_via_block(spec, state)
random_flags(spec, state, 101)
yield from run_process_participation_flag_updates(spec, state)
@with_altair_and_later
@spec_state_test
def test_random_2(spec, state):
next_epoch_via_block(spec, state)
random_flags(spec, state, 102)
yield from run_process_participation_flag_updates(spec, state) yield from run_process_participation_flag_updates(spec, state)