Add int.__rdivmod__

youknowone · youknowone · commit e819b9a2e3cb · 2019-08-13T11:01:28.000+09:00
diff --git a/tests/snippets/ints.py b/tests/snippets/ints.py
@@ -40,6 +40,7 @@
 assert (-2).__divmod__(3) == (-1, 1)
 with assertRaises(ZeroDivisionError):
     (2).__divmod__(0)
+assert (-3).__rdivmod__(2) == (-1, -1)
 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
@@ -415,6 +415,16 @@ impl PyInt {
         }
     }
 
+    #[pymethod(name = "__rdivmod__")]
+    fn rdivmod(&self, other: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+        if objtype::isinstance(&other, &vm.ctx.int_type()) {
+            let other = other.payload::<PyInt>().unwrap();
+            inner_divmod(&other, self, vm)
+        } else {
+            Ok(vm.ctx.not_implemented())
+        }
+    }
+
     #[pymethod(name = "__neg__")]
     fn neg(&self, _vm: &VirtualMachine) -> BigInt {
         -(&self.value)
