Преглед на файлове

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

Sander Mathijs van Veen преди 14 години
родител
ревизия
97b5f9f32f
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  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