Convert BooleanAtom to bool

Author: isuruf

symengine/lib/symengine_wrapper.pyx

@@ -1301,6 +1301,13 @@ cdef class BooleanTrue(BooleanAtom):
     def _sage_(self):
         return True
 
+    def __nonzero__(self):
+        return True
+
+    def __bool__(self):
+        return True
+
+
 true = BooleanTrue()
 
 
@@ -1316,6 +1323,12 @@ cdef class BooleanFalse(BooleanAtom):
     def _sage_(self):
         return False
 
+    def __nonzero__(self):
+        return False
+
+    def __bool__(self):
+        return False
+
 false = BooleanFalse()
 
 

symengine/tests/test_arit.py

@@ -216,3 +216,11 @@ def test_special_constants():
     assert One() == Integer(1)
     assert NegativeOne() == Integer(-1)
     assert Half() == Rational(1, 2)
+
+
+def test_bool():
+    x = Symbol('x')
+    if (x**2).args[1] > 0:
+        assert True
+    if (x**2).args[1] < 0:
+        assert False