Fix +=, -=, and == operators of RuleSize

renatahodovan · Dec 12, 2024 · 2606880 · 2606880
1 parent f5de911
commit 2606880
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/grammarinator/runtime/rule.py b/grammarinator/runtime/rule.py
@@ -53,15 +53,19 @@ def __add__(self, other):
     def __iadd__(self, other):
         self.depth += other.depth
         self.tokens += other.tokens
+        return self
 
     def __sub__(self, other):
         return RuleSize(depth=self.depth - other.depth, tokens=self.tokens - other.tokens)
 
     def __isub__(self, other):
         self.depth -= other.depth
         self.tokens -= other.tokens
+        return self
 
     def __eq__(self, other):
+        if not isinstance(other, RuleSize):
+            return NotImplemented
         return self.depth == other.depth and self.tokens == other.tokens
 
     def __le__(self, other):