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

Fix `test_randomized_state` and `test_randomized_state_leaking`

This commit is contained in:
Hsiao-Wei Wang 2023-05-06 17:45:22 +08:00
parent b617c62e8d
commit 5e2a18a319
No known key found for this signature in database
GPG Key ID: AE3D6B174F971DE4
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/core/pyspec/eth2spec/test/altair/epoch_processing/test_process_inactivity_updates.py
View File

@ -54,7 +54,15 @@ def test_genesis_random_scores(spec, state):
# #
def run_inactivity_scores_test(spec, state, participation_fn=None, inactivity_scores_fn=None, rng=Random(10101)): def run_inactivity_scores_test(spec, state, participation_fn=None, inactivity_scores_fn=None, rng=Random(10101)):
next_epoch_via_block(spec, state) while True:
try:
next_epoch_via_block(spec, state)
except AssertionError:
# If the proposer is slashed, we skip this epoch and try to propose block at the next epoch
next_epoch(spec, state)
else:
break
if participation_fn is not None: if participation_fn is not None:
participation_fn(spec, state, rng=rng) participation_fn(spec, state, rng=rng)
if inactivity_scores_fn is not None: if inactivity_scores_fn is not None:
@ -363,7 +371,7 @@ def test_randomized_state(spec, state):
their inactivity score does not change. their inactivity score does not change.
""" """
rng = Random(10011001) rng = Random(10011001)
_run_randomized_state_test_for_inactivity_updates(spec, state, rng=rng) yield from _run_randomized_state_test_for_inactivity_updates(spec, state, rng=rng)
@with_altair_and_later @with_altair_and_later
@ -377,6 +385,6 @@ def test_randomized_state_leaking(spec, state):
(refer ``get_eligible_validator_indices`). (refer ``get_eligible_validator_indices`).
""" """
rng = Random(10011002) rng = Random(10011002)
_run_randomized_state_test_for_inactivity_updates(spec, state, rng=rng) yield from _run_randomized_state_test_for_inactivity_updates(spec, state, rng=rng)
# Check still in leak # Check still in leak
assert spec.is_in_inactivity_leak(state) assert spec.is_in_inactivity_leak(state)