python - 测试一个类的不同实例是否相等

Question

我正在尝试使用 assert() 来检查一个简单类的内容，而无需检查该类的每个成员。像下面这样的可能吗？

def class class_data(object):
    def __init__ (self, name = 'Richie', school = 'Jefferson High', age = 17):
        self.name = name
        self.school = school
        self.age = age


myschool = class_data()

#check for correct data via assert
assert (myschool == class_data('Fonzie', 'Lincoln High', 17))

如果我的问题毫无意义，请原谅，在此先感谢您。

Answer 1

您需要定义 __eq__类上的方法。

当你执行 myschool == other 时，Python 将调用 myschool.__eq__(other)，默认情况下它只是检查两个对象是否相同(在你的情况下，他们不是)。

您可以覆盖 __eq__ 来实现您的目的，这是一个示例:

def class class_data(object):
    def __init__ (self, name = 'Richie', school = 'Jefferson High', age = 17):
        self.name = name
        self.school = school
        self.age = age

    def __eq__(self, other):
        for attr in ("name", "school", "age"):
            if getattr(self, attr) != getattr(other, attr):
                return False
        return True

注意:在这里使用 getattr 不一定是最好的选择。您选择的实现方式由您决定。