Fix int.__divmod__

youknowone · youknowone · commit f486f3031f48 · 2019-08-13T11:01:28.000+09:00
diff --git a/tests/snippets/builtin_divmod.py b/tests/snippets/builtin_divmod.py
@@ -3,6 +3,7 @@
 assert divmod(11, 3) == (3, 2)
 assert divmod(8,11) == (0, 8)
 assert divmod(0.873, 0.252) == (3.0, 0.11699999999999999)
+assert divmod(-86340, 86400) == (-1, 60)
 
 assert_raises(ZeroDivisionError, lambda: divmod(5, 0), 'divmod by zero')
 assert_raises(ZeroDivisionError, lambda: divmod(5.0, 0.0), 'divmod by zero')
diff --git a/tests/snippets/ints.py b/tests/snippets/ints.py
@@ -37,6 +37,9 @@
 assert (2).__rmul__(1) == 2
 assert (2).__truediv__(1) == 2.0
 assert (2).__rtruediv__(1) == 0.5
+assert (-2).__divmod__(3) == (-1, 1)
+with assertRaises(ZeroDivisionError):
+    (2).__divmod__(0)
 assert (2).__pow__(3) == 8
 assert (10).__pow__(-1) == 0.1
 assert (2).__rpow__(3) == 9
diff --git a/vm/src/obj/objint.rs b/vm/src/obj/objint.rs
@@ -146,6 +146,17 @@ fn inner_mod(int1: &PyInt, int2: &PyInt, vm: &VirtualMachine) -> PyResult {
     }
 }
 
+fn inner_divmod(int1: &PyInt, int2: &PyInt, vm: &VirtualMachine) -> PyResult {
+    if int2.value.is_zero() {
+        Err(vm.new_zero_division_error("integer division or modulo by zero".to_string()))
+    } else {
+        let (div, modulo) = int1.value.div_mod_floor(&int2.value);
+        Ok(vm
+            .ctx
+            .new_tuple(vec![vm.ctx.new_int(div), vm.ctx.new_int(modulo)]))
+    }
+}
+
 #[pyimpl]
 impl PyInt {
     #[pymethod(name = "__eq__")]
@@ -397,15 +408,8 @@ impl PyInt {
     #[pymethod(name = "__divmod__")]
     fn divmod(&self, other: PyObjectRef, vm: &VirtualMachine) -> PyResult {
         if objtype::isinstance(&other, &vm.ctx.int_type()) {
-            let v2 = get_value(&other);
-            if *v2 != BigInt::zero() {
-                let (r1, r2) = self.value.div_rem(v2);
-                Ok(vm
-                    .ctx
-                    .new_tuple(vec![vm.ctx.new_int(r1), vm.ctx.new_int(r2)]))
-            } else {
-                Err(vm.new_zero_division_error("integer divmod by zero".to_string()))
-            }
+            let other = other.payload::<PyInt>().unwrap();
+            inner_divmod(self, &other, vm)
         } else {
             Ok(vm.ctx.not_implemented())
         }
