ソースを参照

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