在pytest中断言时忽略dict中的元素

Question

我想知道在pytest中执行断言时是否有办法忽略dict中的元素.我们有一个断言,它将比较包含last_modified_date的列表.日期将始终更新,因此无法确定日期是否等于最初输入的日期.

例如:

{'userName':'bob','lastModified':'2012-01-01'}

谢谢Jay

Answer 1

ANY系统库中有一个优秀的符号unittest.mock可以用作通配符。尝试这个

from unittest.mock import ANY
actual = {'userName':'bob', 'lastModified':'2012-01-01'}
expected = {'userName':'bob', 'lastModified': ANY}
assert actual == expected

Answer 2

我通过创建等于所有东西的对象解决了这个问题:

class EverythingEquals:
    def __eq__(self, other):
        return True

everything_equals = EverythingEquals()

def test_compare_dicts():
    assert {'userName':'bob','lastModified':'2012-01-01'} == {'userName': 'bob', 'lastModified': everything_equals}

这样它将被比较为相同,你也会检查'lastModified'你的dict.

Answer 3

在断言之前，创建 的副本并从副本中dict删除密钥，或将其设置为静态值。lastModified由于del和dict.update()等不返回dict，您可以为此编写一个辅助函数：

def ignore_keys(d, *args):
    d = dict(d)
    for k in args:
        del d[k]
    return d

assert ignore_keys(myDict, "lastModified") == {"userName": "bob")