소스 검색

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