编写单元测试：我可以在类级别只导入一次我需要的所有模块，还是必须在每次测试中再次导入它们？

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

所以这显然不是正确的方法。

EDIT 只是为了清楚起见，我正在为其编写单元测试的代码（实际上仅由两个方法组成）和（两个）单元测试都在同一个模块中，我们称之为MyCode.py。

Answer 1

鉴于您要删除的重复代码包含导入，我完全同意 BrenBarn 的回应，因为您不需要多次导入模块。 对于您希望在类中的每个测试之前或之后运行相同代码的一般情况，您应该使用类 unittest.TestCase 的 setUp() 和 tearDown() 方法。