断言列表项的相等自定义比较

2022-05-17 00:00:00 python python-unittest

问题描述

在Python中,我想对特定类执行自定义比较。我必须测试的函数返回一个列表,其中包含由我无法控制的库创建的对象。此错误(导致我的测试失败)是由于此特定对象比较造成的。

第一个不同的元素0:

返回实际上是正确的,但库对象的__eq__函数返回False

class NotControlledClass:
    def __eq__(self, other):
        # Always return false as objects are created into the tested function and 
        # the other one into the testing function
        return False

def myCustomEqual(a, b, msg):
    # This is never call
    return a.name == b.name

def test(self):
    self.addTypeEqualityFunc(NotControlledClass, myCustomEqual)

    obj1 = NotControlledClass()
    obj2 = NotControlledClass()

    self.assertEquals([obj1], [obj2])

我尝试遍历结果并逐个比较元素,但问题是该函数可能会将递归结果(如List to List)返回到List。
有人知道如何通过这项测试吗?


解决方案

但问题是该函数可能会将递归结果(如LIST)返回到LIST

所以我们使用递归。我可以推荐以下代码(未经测试):

def assertListNotControlledClassEqual(self, list1, list2, msg=None):
    self.assertEqual(len(list1), len(list2), msg)
    for obj1, obj2 in zip(list1, list2):
        self.assertEqual(type(obj1), type(obj2), msg)
        if isinstance(obj1, list):
            self.assertListNotControlledClassEqual(obj1, obj2, msg)
        else:
            self.assertEqual(obj1, obj2, msg)

...

def test(self):
    self.addTypeEqualityFunc(NotControlledClass, myCustomEqual)

    obj1 = NotControlledClass()
    obj2 = NotControlledClass()

    self.assertListNotControlledClassEqual([obj1], [obj2])

在我看来,这似乎是单元测试中的一个错误。我希望addTypeEqualityFunc向下递归到容器中,但它没有

assertEqual状态手册:

此外,如果第一个和第二个是完全相同的类型,并且是list、tuple、dict、set、Frozenset或str或子类注册的任何类型之一addTypeEqualityFunc(),则将调用特定于类型的相等函数,以便生成更有用的默认错误消息

但是,特定类型的列表方法assertListEqual根本没有提到addTypeEqualityFunc()

相关文章