Implement __eq__ and __ne__ for python Bool (#30)

twizmwazin · Nov 5, 2024 · 4439661 · 4439661
1 parent babcafb
commit 4439661
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/crates/clarirs_py/src/ast/bool.rs b/crates/clarirs_py/src/ast/bool.rs
@@ -300,6 +300,20 @@ impl Bool {
             &GLOBAL_CONTEXT.xor(&self.inner, &<CoerceBool as Into<BoolAst>>::into(other))?,
         )
     }
+
+    pub fn __eq__(&self, py: Python, other: CoerceBool) -> Result<Py<Bool>, ClaripyError> {
+        Bool::new(
+            py,
+            &GLOBAL_CONTEXT.eq_(&self.inner, &<CoerceBool as Into<BoolAst>>::into(other))?,
+        )
+    }
+
+    pub fn __ne__(&self, py: Python, other: CoerceBool) -> Result<Py<Bool>, ClaripyError> {
+        Bool::new(
+            py,
+            &GLOBAL_CONTEXT.neq(&self.inner, &<CoerceBool as Into<BoolAst>>::into(other))?,
+        )
+    }
 }
 
 #[pyfunction(signature = (name, explicit_name = false))]