diff options
Diffstat (limited to 'test/cpp/qps/qps_json_driver_scenario_gen.py')
| -rwxr-xr-x | test/cpp/qps/qps_json_driver_scenario_gen.py | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/test/cpp/qps/qps_json_driver_scenario_gen.py b/test/cpp/qps/qps_json_driver_scenario_gen.py index 28fbb89585..393f968a64 100755 --- a/test/cpp/qps/qps_json_driver_scenario_gen.py +++ b/test/cpp/qps/qps_json_driver_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'] - qps_json_driver_scenario_set = \ - [item for item in all_scenario_set if item['name'] == 'qps_json_driver'] - qps_json_driver_arg_set = \ - [item['args'][2] for item in qps_json_driver_scenario_set \ - if 'args' in item and len(item['args']) > 2] - deserialized_scenarios = [json.loads(item)['scenarios'][0] \ + all_scenario_set = gen.generate_yaml() + all_scenario_set = all_scenario_set['tests'] + qps_json_driver_scenario_set = \ + [item for item in all_scenario_set if item['name'] == 'qps_json_driver'] + qps_json_driver_arg_set = \ + [item['args'][2] for item in qps_json_driver_scenario_set \ + if 'args' in item and len(item['args']) > 2] + deserialized_scenarios = [json.loads(item)['scenarios'][0] \ for item in qps_json_driver_arg_set] - all_scenarios = {scenario['name'].encode('ascii', 'ignore'): \ + 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('qps_json_driver_scenarios.bzl', 'w') as f: - f.write('QPS_JSON_DRIVER_SCENARIOS = ' + serialized_scenarios_str + '\n') + serialized_scenarios_str = str(all_scenarios).encode('ascii', 'ignore') + with open('qps_json_driver_scenarios.bzl', 'w') as f: + f.write('QPS_JSON_DRIVER_SCENARIOS = ' + serialized_scenarios_str + '\n') generate_args() |