Question

我正在学习如何使用Pytest（以及一般的单元测试），我想编写一个测试来检查同一个类的两个对象是否具有相同的属性。

示例：

class Something(object):
    def __init__(self, a, b):
        self.a, self.b = a, b

    def __repr__(self):
        return 'Something(a={}, b={})'.format(self.a, self.b)

def test_equality():
    obj1 = Something(1, 2)
    obj2 = Something(1, 2)
    assert obj1.a == obj2.a
    assert obj1 == obj2

此测试因第三个断言上的AssertionError而失败：

    def test_equality():
        obj1 = Something(1, 2)
        obj2 = Something(1, 2)
        assert obj1.a == obj2.a
        assert obj1.b == obj2.b                                                                                                                                                                                                               
>       assert obj1 == obj2                                                                                                                                                                                                                   
E       assert Something(a=1, b=2) == Something(a=1, b=2)                                                                                                                                                                                     

tests/test_model.py:13: AssertionError

Python或Pytest可以只使用assert obj1 == obj2吗？我应该为每个类实现“丰富的比较”方法，我想用这种方式进行测试还是有一些更简单的方法？

Answer 1

覆盖Something的__eq__函数。

def __eq__(self, other)
    if isinstance(self, other.__class__):
        return self.a == other.a and self.b == other.b
    return False

另外

assert obj1 == obj2

实际上是一个由两部分组成的陈述。首先是表达式obj1 == obj2，它调用obj1 .__ eq __（obj2）并返回一个布尔值，第二个断言为真值的布尔值。

检查两个对象在Pytest中是否具有相同的内容

1 个答案: