fix python

protocolbuffers · Apr 10, 2018 · e3af023 · e3af023
1 parent 76d76ae
commit e3af023
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/benchmarks/python/py_benchmark.py b/benchmarks/python/py_benchmark.py
@@ -144,7 +144,9 @@ def run_benchmark(self, test_method_args='', setup_method_args=''):
       print "Message %s of dataset file %s" % \
           (result["message_name"], result["filename"])
       print "Average time for parse_from_benchmark: %.2f ns" % \
-          (result["benchmarks"]["parse_from_benchmark"])
+          (result["benchmarks"][ \
+                      args.behavior_prefix + "_parse_from_benchmark"])
       print "Average time for serialize_to_benchmark: %.2f ns" % \
-          (result["benchmarks"]["serialize_to_benchmark"])
+          (result["benchmarks"][ \
+                      args.behavior_prefix + "_serialize_to_benchmark"])
       print ""