对python doctests的多版本支持

Question

我正在这样写我的文档测试：

>>> some_function(a=1, b=2)
{u'id': u'123', u'name': u'abc'}

这对于Python 2.5、2.6和2.7版本可以正常工作，但对于Python 3则失败，并出现以下错误：

Expected:
    {u'id': u'123', u'name': u'abc'}
Got:
    {'id': '123', 'name': 'abc'}

问题是，如果我这样编写我的doctest：

>>> some_function(a=1, b=2)
{'id': '123', 'name': 'abc'}

它们仅适用于Python3，而在Python2版本上无效。我的问题是如何使其跨版本兼容？

Answer 1

我在IPython中遇到了与doctests相同的问题。没有整洁的解决方案，但是我将所有u'前缀都包装在中{}，即{u}'，并做了一个小功能，可以酌情包括或排除它们。

您可以看到u_format（）函数和使用它的doctest。

但这很混乱，因此我将许多测试从doctests移开了。

或者，您可以像这样测试它：

>>> some_function(a=1, b=2) == {'id': '123', 'name': 'abc'}
True

如果键中需要一些unicode字符串，则可以使用u'abþ'，并使用distribution2to3在doctest上运行。但这仅适用于输入代码，不适用于输出代表。