summaryrefslogtreecommitdiff
path: root/scripts/internal/bench_oneshot_2.py
blob: 051d00360b3f8fb4934cd9a2b53d22dc2b95b195 (plain)
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"
from bench_oneshot import names

import psutil


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()