Merge pull request RustPython#296 from agude/int_conj

Add .conjugate() method to int type

Author: windelbouwman

tests/snippets/basic_types.py

@@ -45,6 +45,8 @@
 assert type(a) is complex
 assert type(a + a) is complex
 
+a = 1
+assert a.conjugate() == a
 
 a = 12345
 

tests/snippets/bools.py

@@ -46,3 +46,5 @@ def __bool__(self):
 assert False * 7 == 0
 assert True > 0
 assert int(True) == 1
+assert True.conjugate() == 1
+assert isinstance(True.conjugate(), int)

vm/src/obj/objint.rs

@@ -489,6 +489,12 @@ fn int_bit_length(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     Ok(vm.ctx.new_int(bits.to_bigint().unwrap()))
 }
 
+fn int_conjugate(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args, required = [(i, Some(vm.ctx.int_type()))]);
+    let v = get_value(i);
+    Ok(vm.ctx.new_int(v))
+}
+
 pub fn init(context: &PyContext) {
     let ref int_type = context.int_type;
     context.set_attr(&int_type, "__eq__", context.new_rustfunc(int_eq));
@@ -526,4 +532,5 @@ pub fn init(context: &PyContext) {
         "bit_length",
         context.new_rustfunc(int_bit_length),
     );
+    context.set_attr(&int_type, "conjugate", context.new_rustfunc(int_conjugate));
 }