Jelajahi Sumber

person: added equals operator

Fabian Peter Hammerle 4 tahun lalu
induk
melakukan
6e7ae129d1
1 mengubah file dengan 13 tambahan dan 0 penghapusan
  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