断言列表项的相等自定义比较
问题描述
在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()
相关文章