Python 2＆3兼容的Unicode字符串断言相等

我有一些代码返回repr的例外。它需要在Python 2和Python 3的Python 2＆3兼容的Unicode字符串断言相等

运行代码的一个非常简单化下来的版本如下：

from __future__ import unicode_literals 

class Foo: 
    def bar(self): 
     return repr(Exception('bar'))

问题是与测试上面的代码。

的Python 3.5：

foo = Foo() 
assert foo.bar() == "Exception('bar',)" 
# true

的Python 2.7：

foo = Foo() 
assert foo.bar() == "Exception('bar',)" 
# false because foo.bar() returns "Exception(u'bar',)", 
# note the ``u`` before the first ``'``

有没有忽略u一种优雅的方式断言两个字符串的平等是什么时候？我正在使用unittest2。

来源

2016-11-04 zupo

type-cast'bar' to'str'？即'str（'bar'）' –

这听起来像你应该根据正在运行的Python版本不同地定义预期的结果。 – user2357112

这就是我现在拥有的。但是对于多行字符串和很多测试，这意味着很多重复，所以我正在尝试更好的解决方案。 – zupo

没有简单的方法来“改变”字符串相等。有多种方法可以进行近似匹配，但这可能会允许您不需要的其他更改。

解决问题的最佳方法是简单地更改如何定义assert的条件。下面的方案应能在两个python2 & python3工作：

创建动态地期待例外：

assert foo.bar() == repr(Exception(u'bar'))

使用in并为您两种选择：

assert foo.bar() in ("Exception('bar',)", "Exception(u'bar',)")

后者允许u'bar'甚至在python2中，这取决于您的要求，可能是好的或不。

来源

2016-11-04 16:52:44 Bakuriu

Python 2＆3兼容的Unicode字符串断言相等

回答

相关问题