bench/bench.py


1
2
3
4
5
6
7
8
9
10
11
12
13

import sys

if __name__ == "__main__":
    import cProfile
    import pytest  # NOQA
    import pstats

    script = sys.argv[1:] if len(sys.argv) > 1 else "empty.py"
    stats = cProfile.run("pytest.cmdline.main(%r)" % script, "prof")
    p = pstats.Stats("prof")
    p.strip_dirs()
    p.sort_stats("cumulative")
    print(p.print_stats(500))