Implement __eq__ for ParsedMessageCollection

joweich · Nov 27, 2023 · 71990a9 · 71990a9
1 parent e3e201d
commit 71990a9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/chatminer/chatparsers.py b/chatminer/chatparsers.py
@@ -70,6 +70,9 @@ def deserialize_message(mess: dict):
                 deserialize_message(mess) for mess in json.load(json_file)
             ]
 
+    def __eq__(self, other):
+        return self._parsed_messages == other._parsed_messages
+
 
 class Parser(ABC):
     def __init__(self, filepath: str):