intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 22 insertions, 0 deletions

diff --git a/library/python/testing/yatest_common/yatest/common/benchmark.py b/library/python/testing/yatest_common/yatest/common/benchmark.py
new file mode 100644
index 0000000000..c3784cbe4c
--- /dev/null
+++ b/library/python/testing/yatest_common/yatest/common/benchmark.py
@@ -0,0 +1,22 @@
+import json
+
+from . import process
+from . import runtime
+
+
+def execute_benchmark(path, budget=None, threads=None):
+    """
+    Run benchmark and return values
+    :param path: path to benchmark binary
+    :param budget: time budget, sec (supported only by ybenchmark)
+    :param threads: number of threads to run benchmark (supported only by ybenchmark)
+    :return: map of benchmark values
+    """
+    benchmark_path = runtime.binary_path(path)
+    cmd = [benchmark_path, "--benchmark_format=json"]
+    if budget is not None:
+        cmd += ["-b", str(budget)]
+    if threads is not None:
+        cmd += ["-t", str(threads)]
+    res = process.execute(cmd)
+    return json.loads(res.std_out)