Fix tests

This commit is contained in:
Hsiao-Wei Wang 2020-06-10 02:44:09 +08:00
parent dacf86a5c0
commit f0f7bda2ea
No known key found for this signature in database
GPG Key ID: 95B070122902DEA4
2 changed files with 1 additions and 3 deletions

View File

@ -90,7 +90,7 @@ def build_attestation_data(spec, state, slot, index, shard_transition=None, on_t
attestation_data.shard_head_root = shard_transition.shard_data_roots[lastest_shard_data_root_index] attestation_data.shard_head_root = shard_transition.shard_data_roots[lastest_shard_data_root_index]
attestation_data.shard_transition_root = shard_transition.hash_tree_root() attestation_data.shard_transition_root = shard_transition.hash_tree_root()
else: else:
attestation_data.shard_head_root = state.shard_states[shard].transition_digest attestation_data.shard_head_root = state.shard_states[shard].latest_block_root
attestation_data.shard_transition_root = spec.Root() attestation_data.shard_transition_root = spec.Root()
return attestation_data return attestation_data

View File

@ -62,8 +62,6 @@ def get_shard_transitions(spec, parent_beacon_state, shard_blocks):
on_time_slot, on_time_slot,
) )
len_offset_slots = len(offset_slots) len_offset_slots = len(offset_slots)
# TODO this is actually unsafe for long offset_slots
assert len_offset_slots == on_time_slot - parent_beacon_state.shard_states[shard].slot - 1
shard_transition = spec.get_shard_transition(parent_beacon_state, shard, blocks) shard_transition = spec.get_shard_transition(parent_beacon_state, shard, blocks)
if len(blocks) > 0: if len(blocks) > 0: