2

我可以想到两种方法来确保我可以在各种 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？

4

3 回答 3

2

我会使用该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实现。

于 2012-11-12T13:50:17.573 回答
1

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

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

编辑：

事实证明，pyflakes并且flake8对上面的版本不满意，并且会报告“重新定义未使用的'import' from line ...”错误或“W402 'TestCase'导入但未使用”错误。他们似乎更喜欢这样写：

try:
    import unittest2
    TestCase = unittest2.TestCase
except ImportError:
    import unittest
    TestCase = unittest.TestCase
于 2012-11-12T13:14:03.910 回答
0

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

import sys
if sys.version_info < (2, 7):
    import unittest2 as unittest
else:
    import unittest
于 2015-04-20T17:10:53.827 回答