assertEqual列表项的自定义比较

时间:2019-12-27 19:30:48

标签: 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)返回到list。
有人知道如何通过此考试吗?

1 个答案:

答案 0 :(得分:0)

  

但是问题在于该函数可以将递归结果(例如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状态的手册:

  

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

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