summaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
authorDamien George <damien.p.george@gmail.com>2017-08-09 21:25:48 +1000
committerDamien George <damien.p.george@gmail.com>2017-08-09 21:25:48 +1000
commit3d25d9c7d96f9a54ad510391901af8f5bd82b306 (patch)
tree0f3ce56bee2fe8ffebc6ef7ca4fa0bebf1e0d610
parenteb2784e8a2dc17b0f551b68d2b41eece3e5348a7 (diff)
downloadmicropython-3d25d9c7d96f9a54ad510391901af8f5bd82b306.tar.gz
micropython-3d25d9c7d96f9a54ad510391901af8f5bd82b306.zip
py/objstr: Raise an exception for wrong type on RHS of str binary op.
The main case to catch is invalid types for the containment operator, of the form str.__contains__(non-str).
-rw-r--r--py/objstr.c5
-rw-r--r--tests/basics/containment.py11
2 files changed, 14 insertions, 2 deletions
diff --git a/py/objstr.c b/py/objstr.c
index ddad7d3bdf..cc3dda59e5 100644
--- a/py/objstr.c
+++ b/py/objstr.c
@@ -347,8 +347,9 @@ mp_obj_t mp_obj_str_binary_op(mp_uint_t op, mp_obj_t lhs_in, mp_obj_t rhs_in) {
rhs_data = bufinfo.buf;
rhs_len = bufinfo.len;
} else {
- // incompatible types
- return MP_OBJ_NULL; // op not supported
+ // LHS is str and RHS has an incompatible type
+ // (except if operation is EQUAL, but that's handled by mp_obj_equal)
+ bad_implicit_conversion(rhs_in);
}
switch (op) {
diff --git a/tests/basics/containment.py b/tests/basics/containment.py
index bae3661131..4c94a9bae4 100644
--- a/tests/basics/containment.py
+++ b/tests/basics/containment.py
@@ -16,6 +16,17 @@ for needle in [haystack[:i+1] for i in range(len(haystack))]:
print(haystack, "in", needle, "::", haystack in needle)
print(haystack, "not in", needle, "::", haystack not in needle)
+# containment of bytes/ints in bytes
+print(b'' in b'123')
+print(b'0' in b'123', b'1' in b'123')
+print(48 in b'123', 49 in b'123')
+
+# containment of int in str is an error
+try:
+ 1 in '123'
+except TypeError:
+ print('TypeError')
+
# until here, the tests would work without the 'second attempt' iteration thing.
for i in 1, 2: