python - 如何对 python 版本开关进行单元测试

Question

如何为以下函数编写单元测试:

def version_switch():
    if sys.version_info.major < 3:
        print('not python 3')
    else:
        print('python 3')

我的第一个想法是模拟 sys.version_info.major 但因为它是一个只读属性，所以它不会这样做。

可以肯定的是:我确实使用 tox 在 py3 和 py2 下运行我的测试套件。但是，每次运行只会检查一个代码路径。实际功能不使用任何语言特定的功能。我是否应该寻找更容易模拟的替代方法来确定版本？

Answer 1

您可能无法换出 sys.version_info 上的 major 的值，但您可以换出 version_info 的值在 sys 上:

Python 2.7.10 (default, Oct 23 2015, 19:19:21) 
[GCC 4.2.1 Compatible Apple LLVM 7.0.0 (clang-700.0.59.5)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import sys
>>> import mock
>>> with mock.patch.object(sys, 'version_info') as v_info:
...   v_info.major = 3
...   print(sys.version_info.major)
... 
3

请注意，我使用的是 python2.7，但我说服 sys 告诉我我使用的是 python3.x。