Вы должны посмотреть на Doctests . Это очень простой способ научиться тестированию.
По сути, вы пишете свои тесты в интерактивном интерпретаторе, а затем можете копировать / вставлять их в строки документации своих функций.
Пример (из документации по Python)
def factorial(n):
"""Return the factorial of n, an exact integer >= 0.
If the result is small enough to fit in an int, return an int.
Else return a long.
>>> [factorial(n) for n in range(6)]
[1, 1, 2, 6, 24, 120]
>>> [factorial(long(n)) for n in range(6)]
[1, 1, 2, 6, 24, 120]
>>> factorial(30)
265252859812191058636308480000000L
>>> factorial(30L)
265252859812191058636308480000000L
>>> factorial(-1)
Traceback (most recent call last):
...
ValueError: n must be >= 0
Factorials of floats are OK, but the float must be an exact integer:
>>> factorial(30.1)
Traceback (most recent call last):
...
ValueError: n must be exact integer
>>> factorial(30.0)
265252859812191058636308480000000L
It must also not be ridiculously large:
>>> factorial(1e100)
Traceback (most recent call last):
...
OverflowError: n too large
"""
<do stuff>
if __name__ == "__main__":
import doctest
doctest.testmod()
tldr;
Перед строкой кода, которую вы хотите протестировать, вы ставите '>>>', а ожидаемый результат - в строке под ней. Это еще не все, но этого должно быть достаточно, чтобы вы начали.