Profile code executed by pytest
Project description
pytest-line-profiler
line-by-line profiling for code executed by pytest, using line-profiler.
Why?
Line profiler is a wonderful tool to easily identify bottlenecks inside specific functions of your code, and quantify the improvements after a refactor.
Using it is straightforward but required to instrument the functions you want to profile with a "virtual" @profile
decorator
and then execute "a trigger script" (code that calls the decorated functions somehow) via kernprof.py
which works as a python wrapper that understands the decorator, register the functions to be profiled, and print the stats when the script finishes.
Altought it does its job, is a bit invasive: you need to have an special "instrumented" version of your code, and execute it in a way that potentially clashes with the way you do normally (for instance, through a shortcut command from your editor, a test runner, another script, etc.)
Moreover, frequently in real case scenarios, "a trigger script" isn't just a simple function call. You need to prepare input data, connect to external resources, etc. And that's exactly what a test can do, right?
Installation
You can install "pytest-line-profiler" via pip from PyPI.
$ pip install pytest-line-profiler
Usage
Mark your test passing the functions you wants to profile as positional arguments,
like @pytest.mark.line_profile.with_args(function1, function2, [...])
If your test exercises any of those functions, you'll get the profile result as a report.
For example:
import pytest
def f(i):
return i * 10
def g(n=10):
return sum(f(i) for i in range(10))
@pytest.mark.line_profile.with_args(f, g)
def test_as_mark():
assert g() == 450
After that test is executed, you'll get the stats from the line profiler instance.
============ Line Profile result for tests/test_line_profiler.py::test_as_mark ============
Timer unit: 1e-06 s
Total time: 4e-06 s
File: /home/tin/lab/pytest-line-profiler/tests/test_line_profiler.py
Function: f at line 4
Line # Hits Time Per Hit % Time Line Contents
==============================================================
4 def f(i):
5 10 4.0 0.4 100.0 return i * 10
Total time: 3e-05 s
File: /home/tin/lab/pytest-line-profiler/tests/test_line_profiler.py
Function: g at line 7
Line # Hits Time Per Hit % Time Line Contents
==============================================================
7 def g(n=10):
8 1 30.0 30.0 100.0 return sum(f(i) for i in range(10))
Alternatively, you can run any test passing the function/s to profile from the command line
$ pytest --line-profile path.to.function_to_be profiled [...]
Contributing
Contributions are very welcome. Tests can be run with [pytest][https://github.com/pytest-dev/pytest], please ensure the coverage at least stays the same before you submit a pull request.
License
Distributed under the terms of the [MIT][http://opensource.org/licenses/MIT] license, "pytest-line-profiler" is free and open source software
Issues
If you encounter any problems, please [file an issue][https://github.com/mgaitan/pytest-line-profiler/issues] along with a detailed description.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for pytest-line-profiler-0.2.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | b701d72949c9568c7e904de01ba5131ccce540751b0ef06ba409eb5fa551e85d |
|
MD5 | 97dc25a861d0d6e80ae4f2a24fc1ab52 |
|
BLAKE2b-256 | bd05d01f41c43ec9649d4fe378f90539d3101d4ea499ccb07ece8b04cbae259b |
Hashes for pytest_line_profiler-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1995f32f6ee9ab5c331e4c42bc62b58d78875c360fc1d93c5a3ebcf9944abb7f |
|
MD5 | 42def104f87a72cc1a834e106ce08190 |
|
BLAKE2b-256 | 2afe431cb648584d01489d50c2b23f32f59dc7db75fbf1fb2ba48b3df2d6e488 |