Add is_positive, is_negative conditions for Number

Author: isuruf

symengine/lib/symengine.pxd

@@ -273,6 +273,10 @@ cdef extern from "<symengine/symbol.h>" namespace "SymEngine":
 
 cdef extern from "<symengine/number.h>" namespace "SymEngine":
     cdef cppclass Number(Basic):
+        bool is_positive() nogil
+        bool is_negative() nogil
+        bool is_zero() nogil
+        bool is_complex() nogil
         pass
     cdef cppclass NumberWrapper(Basic):
         pass

symengine/lib/symengine_wrapper.pyx

@@ -720,6 +720,34 @@ cdef class Number(Basic):
     def is_Number(self):
         return True
 
+    @property
+    def is_positive(self):
+        return deref(symengine.rcp_static_cast_Number(A.thisptr)).is_positive()
+
+    @property
+    def is_negative(self):
+        return deref(symengine.rcp_static_cast_Number(A.thisptr)).is_negative()
+
+    @property
+    def is_zero(self):
+        return deref(symengine.rcp_static_cast_Number(A.thisptr)).is_zero()
+
+    @property
+    def is_nonzero(self):
+        return not (self.is_complex or self.is_zero)
+
+    @property
+    def is_nonnegative(self):
+        return not (self.is_complex or self.is_negative)
+
+    @property
+    def is_nonpositive(self):
+        return not (self.is_complex or self.is_nonpositive)
+
+    @property
+    def is_complex(self):
+        return deref(symengine.rcp_static_cast_Number(A.thisptr)).is_complex()
+
 cdef class Integer(Number):
 
     @property

symengine/tests/test_integer.py

@@ -0,0 +1,60 @@
+from symengine.utilities import raises
+
+from symengine import Integer, I
+
+def test_integer():
+    i = Integer(5)
+    assert str(i) == "5"
+
+def test_integer_long():
+    i = Integer(123434444444444444444)
+    assert str(i) == "123434444444444444444"
+
+def test_integer_string():
+    assert Integer("133") == 133
+
+def test_smallfloat_valid():
+    i = Integer(7.5)
+    assert str(i) == "7"
+
+def test_bigfloat_valid():
+    i = Integer(13333333333333334.5)
+    assert str(i) == "13333333333333334"
+
+def test_is_conditions():
+    i = Integer(-123)
+    assert not i.is_zero
+    assert not i.is_positive
+    assert i.is_negative
+    assert i.is_nonzero
+    assert i.is_nonpositive
+    assert not i.is_nonnegative
+    assert not i.is_complex
+
+    i = Integer(123)
+    assert not i.is_zero
+    assert i.is_positive
+    assert not i.is_negative
+    assert i.is_nonzero
+    assert not i.is_nonpositive
+    assert i.is_nonnegative
+    assert not i.is_complex
+
+    i = Integer(0)
+    assert i.is_zero
+    assert not i.is_positive
+    assert not i.is_negative
+    assert not i.is_nonzero
+    assert i.is_nonpositive
+    assert i.is_nonnegative
+    assert not i.is_complex
+
+    i = Integer(1) + I
+    assert not i.is_zero
+    assert not i.is_positive
+    assert not i.is_negative
+    assert not i.is_nonzero
+    assert not i.is_nonpositive
+    assert not i.is_nonnegative
+    assert i.is_complex
+