pytest-examples 0.0.13

Description:

pytestexamples 0.0.13

pytest-examples




Pytest plugin for testing Python code examples in docstrings and markdown files.
pytest-examples can:

lint code examples using ruff and black
run code examples
run code examples and check print statements are inlined correctly in the code

It can also update code examples in place to format them and insert or update print statements.
Installation
pip install -U pytest-examples

Usage
Basic usage
Here's an example basic usage - lint then run examples in the foo_dir directory and the bar_file.py file.
import pytest
from pytest_examples import find_examples, CodeExample, EvalExample


@pytest.mark.parametrize('example', find_examples('foo_dir', 'bar_file.py'), ids=str)
def test_docstrings(example: CodeExample, eval_example: EvalExample):
eval_example.lint(example)
eval_example.run(example)

Check print statements
pytest-examples can also check print statements are inserted correctly.
There's the expected format of prints statemints in docstrings:
def add_two_things(a, b):
"""
```py
from my_lib import add_two_things

print(add_two_things(1, 2))
#> 3
```
"""
return a + b

And here's an example of a markdown file, again documenting add_two_things:
# How `add_two_things` works

```py
from my_lib import add_two_things

print(add_two_things(1, 2))
#> 3
```

pytest-examples can then run the code and check the print statements are correct:
import pytest
from pytest_examples import find_examples, CodeExample, EvalExample


@pytest.mark.parametrize('example', find_examples('foo_dir'), ids=str)
def test_docstrings(example: CodeExample, eval_example: EvalExample):
eval_example.run_print_check(example)

Updating files
As well as checking linting and print statements, are correct, we can also update files.
This requires the --update-examples flags AND use of the format() and run_print_update() methods.
Here's a full example of a unit test that checks code when called normally, but can update it
when the flag is set:
import pytest
from pytest_examples import find_examples, CodeExample, EvalExample


@pytest.mark.parametrize('example', find_examples('README.md'), ids=str)
def test_readme(example: CodeExample, eval_example: EvalExample):
if eval_example.update_examples:
eval_example.format(example)
eval_example.run_print_update(example)
else:
eval_example.lint(example)
eval_example.run_print_check(example)