当涉及无序输出时,防止doctest不正确的失败

时间:2015-09-16 11:16:12

标签: python python-3.4 doctest unordered

假设我们有以下功能:

def f(x):
    """
    Turns x into a set.

    >>> given_x = ['a', 'b', 'c', 'd', 'e', 'f']
    >>> f(given_x)
    {'a', 'b', 'c', 'd', 'e', 'f'}

    """

    return set(x)

运行doctest会(通常)导致类似这样的事情:

Failure
**********************************************************************
File "/home/black/Dev/exp_2.py", line 6, in f
Failed example:
    f(given_x)
Expected:
    {'a', 'b', 'c', 'd', 'e', 'f'}
Got:
    {'d', 'e', 'f', 'c', 'a', 'b'}

显然这个失败不应该发生,因为函数按预期工作,但是因为结果是无序的。

我的实际功能输出可能比这复杂得多。它可能是一个带有dicts,sets,list里面的词典。

我需要一个通用的解决方案(如果有的话)。简单sort()对所提出的例子不会解决我的实际案例问题。

问题:
当涉及无序输出时,如何防止doctest(错误地)失败?

2 个答案:

答案 0 :(得分:4)

为什么不直接移动预期的输出,以便测试相等性,预期输出为“True”?

def f(x):
    """
    Turns x into a set. 

    >>> given_x = ['a', 'b', 'c', 'd', 'e', 'f']
    >>> f(given_x) == {'a', 'b', 'c', 'd', 'e', 'f'}
    True
    """

    return set(x)

输出:

Trying:
    given_x = ['a', 'b', 'c', 'd', 'e', 'f']
Expecting nothing
ok
Trying:
    f(given_x) == {'a', 'b', 'c', 'd', 'e', 'f'}
Expecting:
    True
ok

答案 1 :(得分:0)

python设置无法保证顺序,所以你不能依赖它。

我强迫你有一些值得信任的东西:

>>> given_x = ['z','a', 'a', 'b', 'c', 'd', 'e', 'f']
>>> type(f(given_x))
<type 'dict'>
>>> sorted(list(f(given(x))) 
['a', 'b', 'c', 'd', 'e', 'f','z']

我测试它有预期的类型,它实际上做了集合的“唯一性”,结果是我期望的