From 950f2612017c338d5fa4c0fd91193e56bc75e533 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Wed, 26 Sep 2018 15:59:14 -0400 Subject: [PATCH] extract method all_presets --- rl_coach/tests/golden_tests.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/rl_coach/tests/golden_tests.py b/rl_coach/tests/golden_tests.py index 2e61338..654da6f 100644 --- a/rl_coach/tests/golden_tests.py +++ b/rl_coach/tests/golden_tests.py @@ -178,6 +178,13 @@ def perform_reward_based_tests(args, preset_validation_params, preset_name): return test_passed +def all_presets(): + return [ + f[:-3] for f in os.listdir(os.path.join('rl_coach', 'presets')) + if f[-3:] == '.py' and not f == '__init__.py' + ] + + def main(): parser = argparse.ArgumentParser() parser.add_argument('-p', '--preset', @@ -206,8 +213,7 @@ def main(): if args.preset is not None: presets_lists = [args.preset] else: - presets_lists = [f[:-3] for f in os.listdir(os.path.join('rl_coach', 'presets')) if - f[-3:] == '.py' and not f == '__init__.py'] + presets_lists = all_presets() fail_count = 0 test_count = 0