Merge pull request RustPython#1266 from corona10/complex__getnewargs__

complex: Implement __getnewargs__

Author: windelbouwman

tests/snippets/builtin_complex.py

@@ -145,3 +145,7 @@
 assert '(1-1j)' == str(1-1j)
 assert '(1+1j)' == repr(1+1j)
 assert '(1-1j)' == repr(1-1j)
+
+# __getnewargs__
+assert (3 + 5j).__getnewargs__() == (3.0, 5.0)
+assert (5j).__getnewargs__() == (0.0, 5.0)

vm/src/obj/objcomplex.rs

@@ -258,4 +258,12 @@ impl PyComplex {
         let ret = Wrapping(re_hash) + Wrapping(im_hash) * Wrapping(pyhash::IMAG);
         ret.0
     }
+
+    #[pymethod(name = "__getnewargs__")]
+    fn complex_getnewargs(&self, vm: &VirtualMachine) -> PyResult {
+        let Complex64 { re, im } = self.value;
+        Ok(vm
+            .ctx
+            .new_tuple(vec![vm.ctx.new_float(re), vm.ctx.new_float(im)]))
+    }
 }