python - 导入时，为什么我的python unittest脚本会调用我的脚本？

我正在尝试测试python脚本，当我将脚本导入测试套件时，它将调用该脚本。在下面的示例中，我导入list3rdparty，一旦运行测试，它将立即调用list3rdparty。我不希望这种情况发生。我希望测试仅在每个测试用例中调用函数。

list3rdpartytest.py

import unittest
from list3rdparty import * ## this is where the script is being imported


class TestOutputMethods(unittest.TestCase):


    def setUp(self):
        pass

    def test_no_args_returns_help(self):
        args = []
        self.assertEqual(get_third_party(args), help())

    ##get_third_party is a function in list3rdparty##


if __name__ == '__main__':
    unittest.main(warnings = False)

list3rdparty.py

def get_third_party(args_array):
    ##does a bunch of stuff



def get_args():
    get_third_party(sys.argv)

get_args()

最佳答案

您可能具有模块级别的代码，这些代码将在导入时执行。例如，如果您有一个包含以下内容的文件，它将在首次导入时打印该字符串。

import something
from whatever import another

print 'ding'

为了避免这种情况，请将代码放在如下所示的块中：

if __name__ == '__main__':
    # your module-level code here
    get_args()

仅当直接从命令行调用该代码时，它才会运行该代码。