Convert participation_fn from lambda to def
I felt that the the lambda was a little too complicated.
This commit is contained in:
parent
9dcb2eecbc
commit
90c1825637
|
@ -200,7 +200,10 @@ def run_with_participation(spec, state, participation_fn):
|
||||||
@spec_state_test
|
@spec_state_test
|
||||||
def test_almost_empty_attestations(spec, state):
|
def test_almost_empty_attestations(spec, state):
|
||||||
rng = Random(1234)
|
rng = Random(1234)
|
||||||
yield from run_with_participation(spec, state, lambda slot, comm_index, comm: rng.sample(sorted(comm), 1))
|
|
||||||
|
def participation_fn(slot, comm_index, comm):
|
||||||
|
return rng.sample(sorted(comm), 1)
|
||||||
|
yield from run_with_participation(spec, state, participation_fn)
|
||||||
|
|
||||||
|
|
||||||
@with_all_phases
|
@with_all_phases
|
||||||
|
@ -208,15 +211,20 @@ def test_almost_empty_attestations(spec, state):
|
||||||
@leaking()
|
@leaking()
|
||||||
def test_almost_empty_attestations_with_leak(spec, state):
|
def test_almost_empty_attestations_with_leak(spec, state):
|
||||||
rng = Random(1234)
|
rng = Random(1234)
|
||||||
yield from run_with_participation(spec, state, lambda slot, comm_index, comm: rng.sample(sorted(comm), 1))
|
|
||||||
|
def participation_fn(slot, comm_index, comm):
|
||||||
|
return rng.sample(sorted(comm), 1)
|
||||||
|
yield from run_with_participation(spec, state, participation_fn)
|
||||||
|
|
||||||
|
|
||||||
@with_all_phases
|
@with_all_phases
|
||||||
@spec_state_test
|
@spec_state_test
|
||||||
def test_random_fill_attestations(spec, state):
|
def test_random_fill_attestations(spec, state):
|
||||||
rng = Random(4567)
|
rng = Random(4567)
|
||||||
yield from run_with_participation(spec, state,
|
|
||||||
lambda slot, comm_index, comm: rng.sample(sorted(comm), len(comm) // 3))
|
def participation_fn(slot, comm_index, comm):
|
||||||
|
return rng.sample(sorted(comm), len(comm) // 3)
|
||||||
|
yield from run_with_participation(spec, state, participation_fn)
|
||||||
|
|
||||||
|
|
||||||
@with_all_phases
|
@with_all_phases
|
||||||
|
@ -224,16 +232,20 @@ def test_random_fill_attestations(spec, state):
|
||||||
@leaking()
|
@leaking()
|
||||||
def test_random_fill_attestations_with_leak(spec, state):
|
def test_random_fill_attestations_with_leak(spec, state):
|
||||||
rng = Random(4567)
|
rng = Random(4567)
|
||||||
yield from run_with_participation(spec, state,
|
|
||||||
lambda slot, comm_index, comm: rng.sample(sorted(comm), len(comm) // 3))
|
def participation_fn(slot, comm_index, comm):
|
||||||
|
return rng.sample(sorted(comm), len(comm) // 3)
|
||||||
|
yield from run_with_participation(spec, state, participation_fn)
|
||||||
|
|
||||||
|
|
||||||
@with_all_phases
|
@with_all_phases
|
||||||
@spec_state_test
|
@spec_state_test
|
||||||
def test_almost_full_attestations(spec, state):
|
def test_almost_full_attestations(spec, state):
|
||||||
rng = Random(8901)
|
rng = Random(8901)
|
||||||
yield from run_with_participation(spec, state,
|
|
||||||
lambda slot, comm_index, comm: rng.sample(sorted(comm), len(comm) - 1))
|
def participation_fn(slot, comm_index, comm):
|
||||||
|
return rng.sample(sorted(comm), len(comm) - 1)
|
||||||
|
yield from run_with_participation(spec, state, participation_fn)
|
||||||
|
|
||||||
|
|
||||||
@with_all_phases
|
@with_all_phases
|
||||||
|
@ -241,8 +253,10 @@ def test_almost_full_attestations(spec, state):
|
||||||
@leaking()
|
@leaking()
|
||||||
def test_almost_full_attestations_with_leak(spec, state):
|
def test_almost_full_attestations_with_leak(spec, state):
|
||||||
rng = Random(8901)
|
rng = Random(8901)
|
||||||
yield from run_with_participation(spec, state,
|
|
||||||
lambda slot, comm_index, comm: rng.sample(sorted(comm), len(comm) - 1))
|
def participation_fn(slot, comm_index, comm):
|
||||||
|
return rng.sample(sorted(comm), len(comm) - 1)
|
||||||
|
yield from run_with_participation(spec, state, participation_fn)
|
||||||
|
|
||||||
|
|
||||||
@with_all_phases
|
@with_all_phases
|
||||||
|
|
Loading…
Reference in New Issue