在一系列unittest2版本中支持Python特性的最具Python风格的方式是什么？

Question

我可以想到两种方法来确保我可以在各种Python版本中使用unittest库中的现代特性：

try:
    from unittest2 import TestCase
except ImportError:
    from unittest import TestCase

或

import sys
if sys.verson_info.major>=2 and sys.version_info.minor>=7:
    from unittest import TestCase
else:
    from unittest2 import TestCase

其中哪一个更具有Pythonic风格呢？

Answer 1

我会使用try语句。这是一个经常使用的成语。另外，你的sys版本在Python3.3上也是错误的：

>>> if sys.version_info.major>=2 and sys.version_info.minor>=7:
...     from unittest import TestCase
... else:
...     from unittest2 import TestCase
... 
Traceback (most recent call last):
  File "<stdin>", line 4, in <module>
ImportError: No module named 'unittest2'

虽然它应该是：

>>> import sys
>>> major, minor = sys.version_info.major, sys.version_info.minor
>>> if (major >= 2 and minor >= 7) or (major >= 3 and minor >= 2):
...     from unittest import TestCase
... else:
...     from unittest2 import TestCase
... 
>>> 

这也表明try版本比python版本更健壮。

当我有一个用C编写的模块的“加速”版本时，我经常使用try变体，在文件的末尾我放一个：

try:
    from _accelerated import *
except ImportError:
    pass

将python实现改写为被重写的实现。

Answer 2

我不喜欢在第二个版本中我们必须导入另一个模块(sys)，所以我更喜欢第一个版本：

try:
    from unittest2 import TestCase
except ImportError:
    from unittest import TestCase

编辑：

事实证明，pyflakes和flake8对上面的版本并不满意，并将报告“重新定义了行中未使用的‘导入’...”。错误或"W402 'TestCase‘已导入但未使用“错误。他们似乎更喜欢这样写：

try:
    import unittest2
    TestCase = unittest2.TestCase
except ImportError:
    import unittest
    TestCase = unittest.TestCase

Answer 3

对于我们这些避免使用from ... import ...习惯用法的人来说，这将以一种对其余代码透明的方式导入正确的单元测试：

import sys
if sys.version_info < (2, 7):
    import unittest2 as unittest
else:
    import unittest