根据python版本导入`unittest`或`unittest2`的推荐方法是什么

Question

我有这个小test.py脚本，我需要让它在我们的 CI 上运行。 在我们的本地机器中，大多数人使用 python 2.7，因此import unittest可以工作。 但是在 CI 中，环境是 python 2.6，意思是import unittest2 。

目前，我通过调用脚本

if __name__ == '__main__':
    logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(message)s')
    import xmlrunner
    unittest2.main(testRunner=xmlrunner.XMLTestRunner(output='xml-reports'))

Answer 1

您可以在运行时检查您的 python 版本并指定您想要的导入。

import sys

if sys.version_info[0] == 2 and sys.version_info[1] < 7:
    import unittest2
elif sys.version_info[0] == 2 and sys.version_info[1] == 7:
    import unittest

Answer 2

另一种选择是使用 try/except：

try:
    import unittest2 as unittest
except ImportError:
    import unittest

这应该会导致unittest的最佳选项可用于您正在使用的 python，但在所有情况下始终作为unittest导入。