clean up reards generator

This commit is contained in:
Danny Ryan 2020-05-18 20:06:36 -06:00
parent 6d45afeefb
commit 5c564f4d7c
No known key found for this signature in database
GPG Key ID: 2765A792E42CE07A
1 changed files with 9 additions and 9 deletions

View File

@ -14,7 +14,7 @@ from eth2spec.config import config_util
from eth2spec.test.context import PHASE0
def create_provider(handler_name: str, tests_src, config_name: str) -> gen_typing.TestProvider:
def create_provider(tests_src, config_name: str) -> gen_typing.TestProvider:
def prepare_fn(configs_path: str) -> str:
config_util.prepare_config(configs_path, config_name)
@ -25,7 +25,7 @@ def create_provider(handler_name: str, tests_src, config_name: str) -> gen_typin
def cases_fn() -> Iterable[gen_typing.TestCase]:
return generate_from_tests(
runner_name='rewards',
handler_name=handler_name,
handler_name='core',
src=tests_src,
fork_name=PHASE0,
)
@ -34,11 +34,11 @@ def create_provider(handler_name: str, tests_src, config_name: str) -> gen_typin
if __name__ == "__main__":
gen_runner.run_generator("epoch_processing", [
create_provider('get_deltas', test_basic, 'minimal'),
create_provider('get_deltas', test_basic, 'mainnet'),
create_provider('get_deltas', test_leak, 'minimal'),
create_provider('get_deltas', test_leak, 'mainnet'),
create_provider('get_deltas', test_random, 'minimal'),
create_provider('get_deltas', test_random, 'mainnet'),
gen_runner.run_generator("rewards", [
create_provider(test_basic, 'minimal'),
create_provider(test_basic, 'mainnet'),
create_provider(test_leak, 'minimal'),
create_provider(test_leak, 'mainnet'),
create_provider(test_random, 'minimal'),
create_provider(test_random, 'mainnet'),
])