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

update generator type hinting to use new type syntax

This commit is contained in:
protolambda 2019-06-01 01:53:50 +02:00
parent 33233c98ff
commit 77d4f0b407
No known key found for this signature in database
GPG Key ID: EC89FDBB2B4C7623
2 changed files with 20 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
test_libs/pyspec/eth2spec/test/sanity/test_blocks.py
View File

@ -1,4 +1,5 @@
from copy import deepcopy from copy import deepcopy
from typing import List
import eth2spec.phase0.spec as spec import eth2spec.phase0.spec as spec
from eth2spec.utils.bls import bls_sign from eth2spec.utils.bls import bls_sign
@ -36,7 +37,7 @@ def test_empty_block_transition(state):
yield 'pre', state yield 'pre', state
block = build_empty_block_for_next_slot(state, signed=True) block = build_empty_block_for_next_slot(state, signed=True)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -54,7 +55,7 @@ def test_skipped_slots(state):
block = build_empty_block_for_next_slot(state) block = build_empty_block_for_next_slot(state)
block.slot += 3 block.slot += 3
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -72,7 +73,7 @@ def test_empty_epoch_transition(state):
block = build_empty_block_for_next_slot(state) block = build_empty_block_for_next_slot(state)
block.slot += spec.SLOTS_PER_EPOCH block.slot += spec.SLOTS_PER_EPOCH
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -91,7 +92,7 @@ def test_empty_epoch_transition(state):
# block = build_empty_block_for_next_slot(state) # block = build_empty_block_for_next_slot(state)
# block.slot += spec.SLOTS_PER_EPOCH * 5 # block.slot += spec.SLOTS_PER_EPOCH * 5
# sign_block(state, block, proposer_index=0) # sign_block(state, block, proposer_index=0)
# yield 'blocks', [block], [spec.BeaconBlock] # yield 'blocks', [block], List[spec.BeaconBlock]
# #
# state_transition(state, block) # state_transition(state, block)
# yield 'post', state # yield 'post', state
@ -119,7 +120,7 @@ def test_proposer_slashing(state):
block = build_empty_block_for_next_slot(state) block = build_empty_block_for_next_slot(state)
block.body.proposer_slashings.append(proposer_slashing) block.body.proposer_slashings.append(proposer_slashing)
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -152,7 +153,7 @@ def test_attester_slashing(state):
block = build_empty_block_for_next_slot(state) block = build_empty_block_for_next_slot(state)
block.body.attester_slashings.append(attester_slashing) block.body.attester_slashings.append(attester_slashing)
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -189,7 +190,7 @@ def test_deposit_in_block(state):
block.body.deposits.append(deposit) block.body.deposits.append(deposit)
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -216,7 +217,7 @@ def test_deposit_top_up(state):
block.body.deposits.append(deposit) block.body.deposits.append(deposit)
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -252,7 +253,7 @@ def test_attestation(state):
sign_block(state, epoch_block) sign_block(state, epoch_block)
state_transition(state, epoch_block) state_transition(state, epoch_block)
yield 'blocks', [attestation_block, epoch_block], [spec.BeaconBlock] yield 'blocks', [attestation_block, epoch_block], List[spec.BeaconBlock]
yield 'post', state yield 'post', state
assert len(state.current_epoch_attestations) == 0 assert len(state.current_epoch_attestations) == 0
@ -298,7 +299,7 @@ def test_voluntary_exit(state):
sign_block(state, exit_block) sign_block(state, exit_block)
state_transition(state, exit_block) state_transition(state, exit_block)
yield 'blocks', [initiate_exit_block, exit_block], [spec.BeaconBlock] yield 'blocks', [initiate_exit_block, exit_block], List[spec.BeaconBlock]
yield 'post', state yield 'post', state
assert state.validator_registry[validator_index].exit_epoch < spec.FAR_FUTURE_EPOCH assert state.validator_registry[validator_index].exit_epoch < spec.FAR_FUTURE_EPOCH
@ -326,7 +327,7 @@ def test_transfer(state):
block.body.transfers.append(transfer) block.body.transfers.append(transfer)
sign_block(state, block) sign_block(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
state_transition(state, block) state_transition(state, block)
yield 'post', state yield 'post', state
@ -355,7 +356,7 @@ def test_balance_driven_status_transitions(state):
sign_block(state, block) sign_block(state, block)
state_transition(state, block) state_transition(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
yield 'post', state yield 'post', state
assert state.validator_registry[validator_index].exit_epoch < spec.FAR_FUTURE_EPOCH assert state.validator_registry[validator_index].exit_epoch < spec.FAR_FUTURE_EPOCH
@ -371,7 +372,7 @@ def test_historical_batch(state):
block = build_empty_block_for_next_slot(state, signed=True) block = build_empty_block_for_next_slot(state, signed=True)
state_transition(state, block) state_transition(state, block)
yield 'blocks', [block], [spec.BeaconBlock] yield 'blocks', [block], List[spec.BeaconBlock]
yield 'post', state yield 'post', state
assert state.slot == block.slot assert state.slot == block.slot
@ -399,7 +400,7 @@ def test_historical_batch(state):
# #
# state_transition(state, block) # state_transition(state, block)
# #
# yield 'blocks', [block], [spec.BeaconBlock] # yield 'blocks', [block], List[spec.BeaconBlock]
# yield 'post', state # yield 'post', state
# #
# assert state.slot % spec.SLOTS_PER_ETH1_VOTING_PERIOD == 0 # assert state.slot % spec.SLOTS_PER_ETH1_VOTING_PERIOD == 0

9
test_libs/pyspec/eth2spec/test/test_finality.py
View File

@ -1,4 +1,5 @@
from copy import deepcopy from copy import deepcopy
from typing import List
import eth2spec.phase0.spec as spec import eth2spec.phase0.spec as spec
from eth2spec.phase0.spec import ( from eth2spec.phase0.spec import (
@ -86,7 +87,7 @@ def test_finality_rule_4(state):
assert state.finalized_epoch == prev_state.current_justified_epoch assert state.finalized_epoch == prev_state.current_justified_epoch
assert state.finalized_root == prev_state.current_justified_root assert state.finalized_root == prev_state.current_justified_root
yield 'blocks', blocks, [spec.BeaconBlock] yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state yield 'post', state
@ -116,7 +117,7 @@ def test_finality_rule_1(state):
assert state.finalized_epoch == prev_state.previous_justified_epoch assert state.finalized_epoch == prev_state.previous_justified_epoch
assert state.finalized_root == prev_state.previous_justified_root assert state.finalized_root == prev_state.previous_justified_root
yield 'blocks', blocks, [spec.BeaconBlock] yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state yield 'post', state
@ -148,7 +149,7 @@ def test_finality_rule_2(state):
blocks += new_blocks blocks += new_blocks
yield 'blocks', blocks, [spec.BeaconBlock] yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state yield 'post', state
@ -197,5 +198,5 @@ def test_finality_rule_3(state):
assert state.finalized_epoch == prev_state.current_justified_epoch assert state.finalized_epoch == prev_state.current_justified_epoch
assert state.finalized_root == prev_state.current_justified_root assert state.finalized_root == prev_state.current_justified_root
yield 'blocks', blocks, [spec.BeaconBlock] yield 'blocks', blocks, List[spec.BeaconBlock]
yield 'post', state yield 'post', state