1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
|
#!/usr/bin/env python3
# Copyright (c) 2009, Giampaolo Rodola'. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""
Same as bench_oneshot.py but uses perf module instead, which is
supposed to be more precise.
"""
import sys
import pyperf # requires "pip install pyperf"
import psutil
from bench_oneshot import names
p = psutil.Process()
funs = [getattr(p, n) for n in names]
def call_normal():
for fun in funs:
fun()
def call_oneshot():
with p.oneshot():
for fun in funs:
fun()
def add_cmdline_args(cmd, args):
cmd.append(args.benchmark)
def main():
runner = pyperf.Runner()
args = runner.parse_args()
if not args.worker:
print("%s methods involved on platform %r (psutil %s):" % (
len(names), sys.platform, psutil.__version__))
for name in sorted(names):
print(" " + name)
runner.bench_func("normal", call_normal)
runner.bench_func("oneshot", call_oneshot)
main()
|