From 7f27afe83b82f3a98baf58ef595814b9d42a5b2b Mon Sep 17 00:00:00 2001 From: Skye Wanderman-Milne Date: Tue, 30 Jun 2020 01:51:30 -0700 Subject: [PATCH] Expose methods for custom main functions in Python. (#993) --- bindings/python/benchmark/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/bindings/python/benchmark/__init__.py b/bindings/python/benchmark/__init__.py index 27f76e05..378f0304 100644 --- a/bindings/python/benchmark/__init__.py +++ b/bindings/python/benchmark/__init__.py @@ -60,3 +60,8 @@ def _run_benchmarks(argv): def main(argv=None): return app.run(_run_benchmarks, argv=argv, flags_parser=_flags_parser) + + +# Methods for use with custom main function. +initialize = _benchmark.Initialize +run_benchmarks = _benchmark.RunSpecifiedBenchmarks