fix: Integer division issue

Closes #96
radon-project · May 29, 2024 · 22bdec4 · 22bdec4
1 parent 8503174
commit 22bdec4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/core/datatypes.py b/core/datatypes.py
@@ -196,8 +196,7 @@ def idived_by(self, other: Value) -> ResultTuple:
         if isinstance(other, Number):
             if other.value == 0:
                 return None, RTError(other.pos_start, other.pos_end, "Division by zero", self.context)
-
-            return Number(self.value // other.value).set_context(self.context), None
+            return Number(int(self.value // other.value)).set_context(self.context), None
         else:
             return None, Value.illegal_operation(self, other)
 

diff --git a/core/parser.py b/core/parser.py
@@ -478,7 +478,7 @@ def arith_expr(self) -> ParseResult[Node]:
         return self.bin_op(self.term, (TT_PLUS, TT_MINUS))
 
     def term(self) -> ParseResult[Node]:
-        return self.bin_op(self.factor, (TT_MUL, TT_DIV, TT_MOD))
+        return self.bin_op(self.factor, (TT_MUL, TT_DIV, TT_MOD, TT_IDIV))
 
     def factor(self) -> ParseResult[Node]:
         res = ParseResult[Node]()