如果使用__name__ =='__main__'：用於測試

Question

當您編寫一些自足的腳本時，使用if __name__ == '__main__'作爲編寫測試的地方是不是一個好主意？

Answer 1

測試邏輯和測試不應該是「生產」（生產可能意味着您正在使用，發佈到客戶端等）代碼的一部分。因此，在您的腳本中將它們放置在任何地方是個不錯的主意。

理想情況下，將它們放在單獨的文件中。

Answer 2

我想不是，其實我看到很多python腳本（大部分是另一個應用程序的插件）。

Answer 3

最佳做法是將測試放在使用unittest模塊的單獨單元中。這種分離可以讓你保持主代碼的清潔（不需要大量的測試幫助函數），並鼓勵你編寫好的綜合測試，因爲你不會被混淆主代碼所抑制。

Answer 4

這實際上取決於您的代碼和腳本的目的。對於大型複雜的項目，您肯定必須將所有測試放在一個單獨的位置。

但是，當處理一些小事情時，它可能是一個很好的解決方案，可以與代碼一起進行測試 - 它是doctest（它是一個很棒的Python模塊，允許您在文檔字符串中編寫測試）的主要思想。在這種情況下，您if __name__ == '__main__'將是這樣的：

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

我覺得很漂亮和乾淨。