Pythonにはdoctestってのがあるらしい。

26.3. doctest — 対話的な実行例をテストする — Python 3.6.1 ドキュメント

main.py

def add(a, b):
    """
    >>> add(1, 2)
    3
    """
    return a
  

実行

$ python -m doctest main.py -v

まー、失敗

$ python -m doctest main.py -v
Trying:
    add(1, 2)
Expecting:
    3
**********************************************************************
File "/Users/hogehoge/main.py", line 6, in main.add
Failed example:
    add(1, 2)
Expected:
    3
Got:
    1
1 items had no tests:
    main
**********************************************************************
1 items had failures:
   1 of   1 in main.add
1 tests in 2 items.
0 passed and 1 failed.
***Test Failed*** 1 failures.

修正しよう

main.py

def add(a, b):
    """
    >>> add(1, 2)
    3
    """
    return a + b
  

でけた

$ python -m doctest main.py
$ 

-v を付けると合格してもなんか出してくれるらしいよ。 べんりねぇ〜