File: multi_overhead.py

package info (click to toggle)
python-pyinstrument 5.1.1%2Bds-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,624 kB
  • sloc: python: 6,713; ansic: 897; makefile: 46; sh: 26; javascript: 18
file content (102 lines) | stat: -rw-r--r-- 2,310 bytes parent folder | download
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import cProfile
import profile
import re
import sys
import time
from timeit import Timer

import django.conf
import django.template.loader

import pyinstrument

django.conf.settings.configure(
    INSTALLED_APPS=(),
    TEMPLATES=[
        {
            "BACKEND": "django.template.backends.django.DjangoTemplates",
            "DIRS": [
                "./examples/django_example/django_example/templates",
            ],
        }
    ],
)
django.setup()


def test_func_re():
    re.compile(
        r"[a-z0-9!#$%&'*+/=?^_`{|}~-]+(?:\.[a-z0-9!#$%&'*+/=?^_`{|}~-]+)*@(?:[a-z0-9](?:[a-z0-9-]*[a-z0-9])?\.)+[a-z0-9](?:[a-z0-9-]*[a-z0-9])?"
    )


def test_func_template():
    django.template.loader.render_to_string("template.html")


# heat caches
test_func_template()


def time_base(function, repeats):
    timer = Timer(stmt=function)
    return timer.repeat(number=repeats)


def time_profile(function, repeats):
    timer = Timer(stmt=function)
    p = profile.Profile()
    return p.runcall(lambda: timer.repeat(number=repeats))


def time_cProfile(function, repeats):
    timer = Timer(stmt=function)
    p = cProfile.Profile()
    return p.runcall(lambda: timer.repeat(number=repeats))


def time_pyinstrument(function, repeats):
    timer = Timer(stmt=function)
    p = pyinstrument.Profiler()
    p.start()
    result = timer.repeat(number=repeats)
    p.stop()
    return result


profilers = (
    ("Base", time_base),
    # ('profile', time_profile),
    ("cProfile", time_cProfile),
    ("pyinstrument", time_pyinstrument),
)

tests = (
    ("re.compile", test_func_re, 120000),
    ("django template render", test_func_template, 400),
)


def timings_for_test(test_func, repeats):
    results = []
    for profiler_tuple in profilers:
        time = profiler_tuple[1](test_func, repeats)
        results += (profiler_tuple[0], min(time))

    return results


# print header
for column in [""] + [test[0] for test in tests]:
    sys.stdout.write(f"{column:>24}")

sys.stdout.write("\n")

for profiler_tuple in profilers:
    sys.stdout.write(f"{profiler_tuple[0]:>24}")
    sys.stdout.flush()
    for test_tuple in tests:
        time = min(profiler_tuple[1](test_tuple[1], test_tuple[2])) * 10
        sys.stdout.write(f"{time:>24.2f}")
        sys.stdout.flush()
    sys.stdout.write("\n")