# mathx.py
def add(a, b):
    return a + b

# test_mathx.py
from mathx import add

def test_add():
    assert add(2, 3) == 5

Ishga tushirish:

pip install pytest
pytest -q