Add eq method to PySAT model

flamapy · Oct 25, 2024 · e26d9e2 · e26d9e2
1 parent 66d09e1
commit e26d9e2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flamapy/metamodels/pysat_metamodel/models/pysat_model.py b/flamapy/metamodels/pysat_metamodel/models/pysat_model.py
@@ -25,3 +25,6 @@ def get_variable(self, key: str) -> int:
 
     def get_all_clauses(self) -> CNF:
         return self._cnf
+
+    def __eq__(self, other: 'PySATModel') -> bool:
+        return set(self.get_all_clauses().clauses) == set(other.get_all_clauses().clauses)