Add complex.__rmul__

youknowone · youknowone · commit 2a2d0e47648f · 2019-05-04T01:05:27.000+09:00
diff --git a/tests/snippets/builtin_complex.py b/tests/snippets/builtin_complex.py
@@ -29,6 +29,7 @@
 
 assert complex(2, -3) * complex(-5, 7) == complex(11, 29)
 assert complex(2, -3) * 5 == complex(10, -15)
+assert 5 * complex(2, -3) == complex(2, -3) * 5
 
 # __neg__
 
diff --git a/vm/src/obj/objcomplex.rs b/vm/src/obj/objcomplex.rs
@@ -141,6 +141,11 @@ impl PyComplex {
         )
     }
 
+    #[pymethod(name = "__rmul__")]
+    fn rmul(&self, other: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+        self.mul(other, vm)
+    }
+
     #[pymethod(name = "__neg__")]
     fn neg(&self, _vm: &VirtualMachine) -> Complex64 {
         -self.value

Original file line number	Diff line number	Diff line change
`@@ -141,6 +141,11 @@ impl PyComplex {`
`141`	`141`	`)`
`142`	`142`	`}`
`143`	`143`
	`144`	`+ #[pymethod(name = "__rmul__")]`
	`145`	`+ fn rmul(&self, other: PyObjectRef, vm: &VirtualMachine) -> PyResult {`
	`146`	`+ self.mul(other, vm)`
	`147`	`+ }`
	`148`	`+`
`144`	`149`	`#[pymethod(name = "__neg__")]`
`145`	`150`	`fn neg(&self, _vm: &VirtualMachine) -> Complex64 {`
`146`	`151`	`-self.value`