根据python版本的不同，导入`unittest`或`unittest2`的推荐方式是什么

Question

我有一个小test.py脚本，我需要让它在我们的CI上运行。在我们的本地机器上，大多数人使用Python2.7，所以import unittest可以工作。但是在CI中，环境是Python2.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

这应该会为您正在使用的python提供最佳的unittest选项，但在所有情况下都会一致地导入为unittest。