doctest使用示例

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

    >>> add([1], [2])
    [1, 2]

    >>> add("1", "2")
    '12'
    """
    return a + b


if __name__ == "__main__":
    import doctest
    doctest.testmod()

 可以直接在命令行进行测试 python -m doctest -v example.py 会输出详细测试结果