From 5c564f4d7c9463e553641fc1792f879a2aadecb5 Mon Sep 17 00:00:00 2001 From: Danny Ryan Date: Mon, 18 May 2020 20:06:36 -0600 Subject: [PATCH] clean up reards generator --- tests/generators/rewards/main.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/generators/rewards/main.py b/tests/generators/rewards/main.py index d8dae74fa..c90943cab 100644 --- a/tests/generators/rewards/main.py +++ b/tests/generators/rewards/main.py @@ -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'), ])