diff options
Diffstat (limited to 'test/cpp/qps/json_run_localhost_scenario_gen.py')
-rwxr-xr-x | test/cpp/qps/json_run_localhost_scenario_gen.py | 30 |
1 files changed, 15 insertions, 15 deletions
diff --git a/test/cpp/qps/json_run_localhost_scenario_gen.py b/test/cpp/qps/json_run_localhost_scenario_gen.py index 3cd2b799fe..01f6a699d5 100755 --- a/test/cpp/qps/json_run_localhost_scenario_gen.py +++ b/test/cpp/qps/json_run_localhost_scenario_gen.py @@ -18,21 +18,21 @@ import gen_build_yaml as gen import json def generate_args(): - all_scenario_set = gen.generate_yaml() - all_scenario_set = all_scenario_set['tests'] - json_run_localhost_scenarios = \ - [item for item in all_scenario_set if item['name'] == 'qps_json_driver'] - json_run_localhost_arg_set = \ - [item['args'][2] for item in json_run_localhost_scenarios \ - if 'args' in item and len(item['args']) > 2] - deserialized_scenarios = [json.loads(item)['scenarios'][0] \ - for item in json_run_localhost_arg_set] - all_scenarios = {scenario['name'].encode('ascii', 'ignore'): \ - '\'{\'scenarios\' : [' + json.dumps(scenario) + ']}\'' \ - for scenario in deserialized_scenarios} + all_scenario_set = gen.generate_yaml() + all_scenario_set = all_scenario_set['tests'] + json_run_localhost_scenarios = \ + [item for item in all_scenario_set if item['name'] == 'qps_json_driver'] + json_run_localhost_arg_set = \ + [item['args'][2] for item in json_run_localhost_scenarios \ + if 'args' in item and len(item['args']) > 2] + deserialized_scenarios = [json.loads(item)['scenarios'][0] \ + for item in json_run_localhost_arg_set] + all_scenarios = {scenario['name'].encode('ascii', 'ignore'): \ + '\'{\'scenarios\' : [' + json.dumps(scenario) + ']}\'' \ + for scenario in deserialized_scenarios} - serialized_scenarios_str = str(all_scenarios).encode('ascii', 'ignore') - with open('json_run_localhost_scenarios.bzl', 'wb') as f: - f.write('JSON_RUN_LOCALHOST_SCENARIOS = ' + serialized_scenarios_str + '\n') + serialized_scenarios_str = str(all_scenarios).encode('ascii', 'ignore') + with open('json_run_localhost_scenarios.bzl', 'wb') as f: + f.write('JSON_RUN_LOCALHOST_SCENARIOS = ' + serialized_scenarios_str + '\n') generate_args() |