pytest/bench/bench.py

18 lines
383 B
Python

from __future__ import annotations
import sys
if __name__ == "__main__":
import cProfile
import pstats
import pytest # noqa: F401
script = sys.argv[1:] if len(sys.argv) > 1 else ["empty.py"]
cProfile.run(f"pytest.cmdline.main({script!r})", "prof")
p = pstats.Stats("prof")
p.strip_dirs()
p.sort_stats("cumulative")
print(p.print_stats(500))