Profile
Back to NewsBack
GitHub Trending 5 min
Reader Mode
ionelmc/pytest-benchmark: pytest fixture for benchmarking code

ionelmc/pytest-benchmark: pytest fixture for benchmarking code

6 hours ago

======== Overview ========

.. list-table:: :stub-columns: 1

* - docs - |docs| |gitter| * - tests - |github-actions| |coveralls| |codecov| * - package - |version| |wheel| |supported-versions| |supported-implementations| |commits-since| .. |docs| image:: https://readthedocs.org/projects/pytest-benchmark/badge/?style=flat :target: https://readthedocs.org/projects/pytest-benchmark/ :alt: Documentation Status .. |github-actions| image:: https://github.com/ionelmc/pytest-benchmark/actions/workflows/github-actions.yml/badge.svg :alt: GitHub Actions Build Status :target: https://github.com/ionelmc/pytest-benchmark/actions .. |gitter| image:: https://badges.gitter.im/ionelmc/pytest-benchmark.svg :alt: Join the chat at https://gitter.im/ionelmc/pytest-benchmark :target: https://gitter.im/ionelmc/pytest-benchmark .. |coveralls| image:: https://coveralls.io/repos/github/ionelmc/pytest-benchmark/badge.svg?branch=master :alt: Coverage Status :target: https://coveralls.io/github/ionelmc/pytest-benchmark?branch=master .. |codecov| image:: https://codecov.io/gh/ionelmc/pytest-benchmark/branch/master/graphs/badge.svg?branch=master :alt: Coverage Status :target: https://app.codecov.io/github/ionelmc/pytest-benchmark .. |version| image:: https://img.shields.io/pypi/v/pytest-benchmark.svg :alt: PyPI Package latest release :target: https://pypi.org/project/pytest-benchmark .. |wheel| image:: https://img.shields.io/pypi/wheel/pytest-benchmark.svg :alt: PyPI Wheel :target: https://pypi.org/project/pytest-benchmark .. |supported-versions| image:: https://img.shields.io/pypi/pyversions/pytest-benchmark.svg :alt: Supported versions :target: https://pypi.org/project/pytest-benchmark .. |supported-implementations| image:: https://img.shields.io/pypi/implementation/pytest-benchmark.svg :alt: Supported implementations :target: https://pypi.org/project/pytest-benchmark .. |commits-since| image:: https://img.shields.io/github/commits-since/ionelmc/pytest-benchmark/v5.3.0.svg :alt: Commits since latest release :target: https://github.com/ionelmc/pytest-benchmark/compare/v5.3.0...master

A `pytest fixture for benchmarking code. It will group the tests into rounds that are calibrated to the chosen timer.

See calibration_ and FAQ_.

  • Free software: BSD 2-Clause License
Installation ============

::

pip install pytest-benchmark

Documentation =============

For latest release: pytest-benchmark.readthedocs.org/en/stable _.

For master branch (may include documentation fixes): pytest-benchmark.readthedocs.io/en/latest _.

Examples ========

But first, a prologue:

This plugin tightly integrates into pytest. To use this effectively you should know a thing or two about pytest first. Take a look at the introductory material _ or watch talks _.

Few notes:

* This plugin benchmarks functions and only that. If you want to measure block of code or whole programs you will need to write a wrapper function. * In a test you can only benchmark one function. If you want to benchmark many functions write more tests or use parametrization _. * To run the benchmarks you simply use pytest to run your "tests". The plugin will automatically do the benchmarking and generate a result table. Run pytest --help for more details.

This plugin provides a benchmark fixture. This fixture is a callable object that will benchmark any function passed to it.

Example:

.. code-block:: python

def something(duration=0.000001): """ Function that needs some serious benchmarking. """ time.sleep(duration) # You may return anything you want, like the result of a computation return 123

def test_my_stuff(benchmark): # benchmark something result = benchmark(something)

# Extra code, to verify that the run completed correctly. # Sometimes you may want to check the result, fast functions # are no good if they return incorrect results :-) assert result == 123

You can also pass extra arguments:

.. code-block:: python

def test_my_stuff(benchmark): benchmark(time.sleep, 0.02)

Or even keyword arguments:

.. code-block:: python

def test_my_stuff(benchmark): benchmark(time.sleep, duration=0.02)

Another pattern seen in the wild, that is not recommended for micro-benchmarks (very fast code) but may be convenient:

.. code-block:: python

def test_my_stuff(benchmark): @benchmark def something(): # unnecessary function call time.sleep(0.000001)

A better way is to just benchmark the final function:

.. code-block:: python

def test_my_stuff(benchmark): benchmark(time.sleep, 0.000001) # way more accurate results!

If you need to do fine control over how the benchmark is run (like a setup function, exact control of iterations and rounds) there's a special mode - pedantic_:

.. code-block:: python

def my_special_setup(): ...

def test_with_setup(benchmark): benchmark.pedantic(something, setup=my_special_setup, args=(1, 2, 3), kwargs={'foo': 'bar'}, iterations=10, rounds=100)

Screenshots -----------

Normal run:

.. image:: https://github.com/ionelmc/pytest-benchmark/raw/master/docs/screenshot.png :alt: Screenshot of pytest summary

Compare mode (--benchmark-compare):

.. image:: https://github.com/ionelmc/pytest-benchmark/raw/master/docs/screenshot-compare.png :alt: Screenshot of pytest summary in compare mode

Histogram (--benchmark-histogram):

.. image:: https://cdn.rawgit.com/ionelmc/pytest-benchmark/94860cc8f47aed7ba4f9c7e1380c2195342613f6/docs/sample-tests_test_normal.py_test_xfast_parametrized%5B0%5D.svg :alt: Histogram sample

..

Also, it has nice tooltips _.

Compare between mode (pytest-benchmark compare --between`):

.. image:: https://github.com/ionelmc/pytest-benchmark/raw/master/docs/screenshot-compare-between.png :alt: Screenshot of pytest summary in compare between mode

Development ===========

To run the all tests run::

tox

Credits =======

  • Timing code and ideas taken from: https://github.com/vstinner/misc/blob/34d3128468e450dad15b6581af96a790f8bd58ce/python/benchmark.py
.. _FAQ: http://pytest-benchmark.readthedocs.org/en/latest/faq.html .. _calibration: http://pytest-benchmark.readthedocs.org/en/latest/calibration.html .. _pedantic: http://pytest-benchmark.readthedocs.org/en/latest/pedantic.html
Chat with me