summaryrefslogtreecommitdiff
path: root/py/objcomplex.c
diff options
context:
space:
mode:
authorPaul Sokolovsky <pfalcon@users.sourceforge.net>2014-01-30 04:37:19 +0200
committerPaul Sokolovsky <pfalcon@users.sourceforge.net>2014-01-30 04:37:19 +0200
commitc1d9bbc3453454aceb28f51e72e4aeb8ef1c12eb (patch)
treedb7cd93c99583a858ef94b3dab95b882832da251 /py/objcomplex.c
parentcdd2c62e07549e36dba00bc37d7ba7a4cd41ad50 (diff)
Implement __bool__ and __len__ via unary_op virtual method for all types.
__bool__() and __len__() are just the same as __neg__() or __invert__(), and require efficient dispatching implementation (not requiring search/lookup). type->unary_op() is just the right choice for this short of adding standalone virtual method(s) to already big mp_obj_type_t structure.
Diffstat (limited to 'py/objcomplex.c')
-rw-r--r--py/objcomplex.c2
1 files changed, 1 insertions, 1 deletions
diff --git a/py/objcomplex.c b/py/objcomplex.c
index 24762e8b1..5b4df0da6 100644
--- a/py/objcomplex.c
+++ b/py/objcomplex.c
@@ -73,7 +73,7 @@ static mp_obj_t complex_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const
static mp_obj_t complex_unary_op(int op, mp_obj_t o_in) {
mp_obj_complex_t *o = o_in;
switch (op) {
- case RT_UNARY_OP_NOT: if (o->real != 0 || o->imag != 0) { return mp_const_true;} else { return mp_const_false; }
+ case RT_UNARY_OP_BOOL: return MP_BOOL(o->real != 0 || o->imag != 0);
case RT_UNARY_OP_POSITIVE: return o_in;
case RT_UNARY_OP_NEGATIVE: return mp_obj_new_complex(-o->real, -o->imag);
default: return MP_OBJ_NULL; // op not supported