From bfca7f9a96a6847940c73bd2508b18318577de45 Mon Sep 17 00:00:00 2001 From: Danny Ryan Date: Wed, 21 Sep 2022 07:33:05 -0600 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Alex Stokes --- .../test/capella/block_processing/test_process_deposit.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/core/pyspec/eth2spec/test/capella/block_processing/test_process_deposit.py b/tests/core/pyspec/eth2spec/test/capella/block_processing/test_process_deposit.py index 93dd655fc..e0603d301 100644 --- a/tests/core/pyspec/eth2spec/test/capella/block_processing/test_process_deposit.py +++ b/tests/core/pyspec/eth2spec/test/capella/block_processing/test_process_deposit.py @@ -20,6 +20,9 @@ def test_success_top_up_to_withdrawn_validator(spec, state): assert state.balances[validator_index] > 0 next_epoch_via_block(spec, state) assert state.balances[validator_index] == 0 + assert state.validators[validator_index].effective_balance > 0 + next_epoch_via_block(spec, state) + assert state.validators[validator_index].effective_balance == 0 # Make a top-up balance to validator amount = spec.MAX_EFFECTIVE_BALANCE // 4 @@ -27,8 +30,8 @@ def test_success_top_up_to_withdrawn_validator(spec, state): yield from run_deposit_processing(spec, state, deposit, validator_index) - state.balances[validator_index] == amount - state.validators[validator_index].effective_balance == 0 + assert state.balances[validator_index] == amount + assert state.validators[validator_index].effective_balance == 0 validator = state.validators[validator_index] balance = state.balances[validator_index]