Какой тип возвращает метод __eq__ в Python

Что проверяют

Знаете ли вы протокол сравнения Python глубже, чем «вернёт True или False». Хороший ответ содержит NotImplemented, а следом обычно спрашивают, что станет с __hash__, когда вы переопределили __eq__.

Как отвечать

Формально __eq__ может вернуть любой объект, по соглашению возвращает bool. Важны три случая:

• bool: обычный результат, когда сравнение для этой пары типов определено;
• NotImplemented: специальный синглтон, который означает «не умею сравнивать себя с этим типом». Python тогда пробует отражённый вызов other.__eq__(self), а если и он вернул NotImplemented, откатывается к сравнению по идентичности, то есть a is b;
• произвольный объект: так делают библиотеки. У NumPy a == b возвращает массив булевых значений поэлементно, у SQLAlchemy User.id == 5 возвращает выражение, из которого собирается WHERE.

Типовая реализация начинается с if not isinstance(other, Point): return NotImplemented, дальше сравнение полей. Именно NotImplemented, а не False: второй операнд получает шанс сравнить себя сам, и сравнение с подклассом работает симметрично.

Ещё два факта, которые стоит назвать сразу:

• __ne__: в Python 3 по умолчанию инвертирует результат __eq__, если это не NotImplemented, отдельно его определять не нужно;
• __hash__: если в классе определён __eq__, а __hash__ нет, Python выставляет __hash__ = None. Объекты перестают быть хэшируемыми, их нельзя положить в set или сделать ключом dict. @dataclass генерирует __eq__ и ведёт себя так же, пока не указан frozen=True или unsafe_hash=True.

Где спотыкаются

Говорят «всегда bool». Для большинства своих классов это так, но пример с NumPy, где if a == b: падает с ValueError про неоднозначную истинность массива, показывает, что гарантии нет.

Возвращают False для чужих типов. Сравнение перестаёт быть симметричным, а отражённый метод второго операнда вообще не вызывается.

Путают NotImplemented с NotImplementedError. Первое возвращают из методов сравнения, второе бросают из абстрактных методов, и raise NotImplementedError в __eq__ просто уронит сравнение.