소스 검색

person: added equals operator

Fabian Peter Hammerle 5 년 전
부모
커밋
6e7ae129d1
1개의 변경된 파일13개의 추가작업 그리고 0개의 파일을 삭제
  1. 13 0
      family_tree_yaml/__init__.py

+ 13 - 0
family_tree_yaml/__init__.py

@@ -54,3 +54,16 @@ class Person:
         return (self.name or "unnamed") + (
             " (*{})".format(self.birth_date.isoformat()) if self.birth_date else ""
         )
+
+    def __eq__(self, other: "Person") -> bool:
+        """
+        >>> maxl = Person("max")
+        >>> maxl.name = "Max Mustermann"
+        >>> maxl == Person("max")
+        True
+        >>> erika = Person("erika")
+        >>> erika.name = "Max Mustermann"
+        >>> maxl == erika
+        False
+        """
+        return self.identifier == other.identifier