eth2.0-specs/test_libs/pyspec/eth2spec/testing/context.py

22 lines
574 B
Python

from eth2spec.phase0 import spec
from .helpers import create_genesis_state
from .utils import spectest, with_args
# Provides a genesis state as first argument to the function decorated with this
with_state = with_args(lambda: [create_genesis_state(spec.SLOTS_PER_EPOCH * 8, list())])
# shorthand for decorating @with_state @spectest()
def spec_state_test(fn):
return with_state(spectest()(fn))
def expect_assertion_error(fn):
try:
fn()
raise AssertionError('expected an assertion error, but got none.')
except AssertionError:
pass