python - 使用 if __name__ == '__main__' : for tests

Question

当您编写一些自给自足的脚本时，使用 if __name__ == '__main__' 作为编写测试的地方是不是一个坏主意？

Answer 1

这实际上取决于您的代码和脚本的目的。对于大型和复杂的项目，您肯定必须将所有测试放在一个单独的地方。

但是在处理一些小的事情时，将测试与代码一起进行可能是一个很好的解决方案 - 这是 doctest 的主要思想(这是一个很棒的 Python 模块，允许您在文档字符串中编写测试)。在这种情况下，您的 if __name__ == '__main__' 将如下所示:

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

我觉得它很好很干净。