Adding 'int.__bool__' method

mrveera · mrveera · commit 0c565e96c497 · 2019-02-05T00:42:45.000+05:30
diff --git a/tests/snippets/numbers.py b/tests/snippets/numbers.py
@@ -8,3 +8,5 @@ class A(int):
 x = A(7)
 assert x == 7
 assert type(x) is A
+
+assert int(2).__bool__() == True
diff --git a/vm/src/obj/objint.rs b/vm/src/obj/objint.rs
@@ -90,6 +90,10 @@ impl FromPyObjectRef for BigInt {
     }
 }
 
+fn int_bool(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args, required = [(zelf, Some(vm.ctx.int_type()))]);
+    Ok(vm.ctx.new_bool(true))
+}
 fn int_eq(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(
         vm,
@@ -527,6 +531,7 @@ pub fn init(context: &PyContext) {
     context.set_attr(&int_type, "__format__", context.new_rustfunc(int_format));
     context.set_attr(&int_type, "__truediv__", context.new_rustfunc(int_truediv));
     context.set_attr(&int_type, "__xor__", context.new_rustfunc(int_xor));
+    context.set_attr(&int_type, "__bool__", context.new_rustfunc(int_bool));
     context.set_attr(
         &int_type,
         "bit_length",
