diff options
| -rw-r--r-- | py/objtype.c | 4 | ||||
| -rw-r--r-- | tests/basics/class-contains.py | 23 |
2 files changed, 26 insertions, 1 deletions
diff --git a/py/objtype.c b/py/objtype.c index f96a29beb8..80f39d7f1f 100644 --- a/py/objtype.c +++ b/py/objtype.c @@ -384,7 +384,9 @@ STATIC const qstr binary_op_method_name[] = { MP_BINARY_OP_LESS_EQUAL, MP_BINARY_OP_MORE_EQUAL, MP_BINARY_OP_NOT_EQUAL, - MP_BINARY_OP_IN, + */ + [MP_BINARY_OP_IN] = MP_QSTR___contains__, + /* MP_BINARY_OP_IS, */ [MP_BINARY_OP_EXCEPTION_MATCH] = MP_QSTR_, // not implemented, used to make sure array has full size diff --git a/tests/basics/class-contains.py b/tests/basics/class-contains.py new file mode 100644 index 0000000000..b6dd3661cd --- /dev/null +++ b/tests/basics/class-contains.py @@ -0,0 +1,23 @@ +# A contains everything +class A: + def __contains__(self, key): + return True + +a = A() +print(True in a) +print(1 in a) +print(() in a) + +# B contains given things +class B: + def __init__(self, items): + self.items = items + def __contains__(self, key): + return key in self.items + +b = B([]) +print(1 in b) +b = B([1, 2]) +print(1 in b) +print(2 in b) +print(3 in b) |