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

97b5f9f3 · Sander Mathijs van Veen · 26b4d442 · 97b5f9f3
Commit 97b5f9f3 authored Jan 05, 2012 by Sander Mathijs van Veen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/node.py src/node.py +3 -0

No files found.
--- a/src/node.py
+++ b/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