test_ok2/bench/bench.py

13 lines
336 B
Python

import sys
if __name__ == '__main__':
import cProfile
import pytest
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))