Explorar el Código

Added is_op to strictly check if a node is a certain operator.

Sander Mathijs van Veen hace 14 años
padre
commit
97b5f9f32f
Se han modificado 1 ficheros con 3 adiciones y 0 borrados
  1. 3 0
      src/node.py

+ 3 - 0
src/node.py

@@ -91,6 +91,9 @@ class ExpressionBase(object):
 
         return s_root < o_root or s_exp < o_exp or s_coeff < o_coeff
 
+    def is_op(self, op):
+        return not self.is_leaf() and self.op == op
+
     def is_leaf(self):
         return self.type != TYPE_OPERATOR