編寫單元測試：我可以在課堂級別導入我需要的所有模塊，還是必須在每次測試中再次導入它們？

Question

我想要編寫單元測試使用unittest一些代碼：

https://docs.python.org/3/library/unittest.html

假設每個測試我寫需要導入math，os和datetime模塊。現在，我輸入他們在每個我寫的測試：

#...code for which I'm writing the unit tests... 
import unittest 

class TestMyCode(unittest.TestCase): 

    def test_method_1(self): 
     # unit test for method 1 
     import math 
     import os 
     import datetime 
     . 
     . 

    def test_method_2(self): 
     # unit test for method 2 
     import math 
     import os 
     import datetime 
     . 
     . 

if __name__ == "__main__": 
    unittest.main() 

爲了避免重複代碼，是不是有可能在類級別只導入一次？這：

#...code for which I'm writing the unit tests... 
import unittest 

class TestMyCode(unittest.TestCase): 

    import math 
    import os 
    import datetime 

    def test_method_1(self): 
     # unit test for method 1 
     . 
     . 

    def test_method_2(self): 
     # unit test for method 2 
     . 
     . 

if __name__ == "__main__": 
    unittest.main() 

導致錯誤

NameError: name 'math' is not defined 

因此，這顯然不是正確的做法。爲了清楚起見，我正在編寫單元測試（實際上由兩個方法組成）和（兩個）單元測試的代碼都在同一個模塊中，我們稱之爲MyCode.py。

Answer 1

鑑於您想要擺脫的重複代碼包含導入，我完全同意BrenBarn的迴應，因爲您不需要多次導入模塊。 對於要在類中的每個測試之前或之後運行相同代碼的一般情況，您應該使用class unittest.TestCase的setUp（）和tearDown（）方法。