瀏覽代碼

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