Add mul method for the complex type

Author: Nicolas Trinquier

tests/snippets/builtin_complex.py

@@ -24,6 +24,11 @@
 assert complex(1, 2) != 'foo'
 assert complex(1, 2).__eq__('foo') == NotImplemented
 
+# __mul__
+
+assert complex(2, -3) * complex(-5, 7) == complex(-21, 29)
+assert complex(2, -3) * 5 == complex(10, -15)
+
 # __neg__
 
 assert -complex(1, -1) == complex(-1, 1)

vm/src/obj/objcomplex.rs

@@ -161,6 +161,18 @@ impl PyComplex {
         vm.ctx.new_bool(result)
     }
 
+    #[pymethod(name = "__mul__")]
+    fn mul(self, other: PyObjectRef, vm: &VirtualMachine) -> PyObjectRef {
+        match to_complex(other, vm) {
+            Ok(Some(other)) => Ok(vm.ctx.new_complex(Complex64::new(
+                self.value.re * other.re - self.value.im * other.im,
+                self.value.re * other.im + self.value.re * other.im,
+            ))),
+            Ok(None) => Ok(vm.ctx.not_implemented()),
+            Err(err) => Err(err),
+        }
+    }
+
     #[pymethod(name = "__neg__")]
     fn neg(&self, _vm: &VirtualMachine) -> PyComplex {
         PyComplex::from(-self.value)