Implement tuple.index().

2 files changed, 17 insertions, 3 deletions

diff --git a/py/objtuple.c b/py/objtuple.c
index 18eb6df4de..581bfb6290 100644
--- a/py/objtuple.c
+++ b/py/objtuple.c
@@ -153,6 +153,18 @@ static mp_obj_t tuple_getiter(mp_obj_t o_in) {
     return mp_obj_new_tuple_iterator(o_in, 0);
 }
 
+static mp_obj_t tuple_index(uint n_args, const mp_obj_t *args) {
+    assert(MP_OBJ_IS_TYPE(args[0], &tuple_type));
+    mp_obj_tuple_t *self = args[0];
+    return mp_seq_index_obj(self->items, self->len, n_args, args);
+}
+static MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(tuple_index_obj, 2, 4, tuple_index);
+
+static const mp_method_t tuple_type_methods[] = {
+    { "index", &tuple_index_obj },
+    { NULL, NULL }, // end-of-list sentinel
+};
+
 const mp_obj_type_t tuple_type = {
     { &mp_const_type },
     "tuple",
@@ -161,6 +173,7 @@ const mp_obj_type_t tuple_type = {
     .unary_op = tuple_unary_op,
     .binary_op = tuple_binary_op,
     .getiter = tuple_getiter,
+    .methods = tuple_type_methods,
 };
 
 // the zero-length tuple
diff --git a/py/sequence.c b/py/sequence.c
index 07c0531b10..b3fb9fcd34 100644
--- a/py/sequence.c
+++ b/py/sequence.c
@@ -156,9 +156,10 @@ mp_obj_t mp_seq_index_obj(const mp_obj_t *items, uint len, uint n_args, const mp
     }
 
     for (uint i = start; i < stop; i++) {
-         if (mp_obj_equal(items[i], value)) {
-              return mp_obj_new_int_from_uint(i);
-         }
+        if (mp_obj_equal(items[i], value)) {
+            // Common sense says this cannot overflow small int
+            return MP_OBJ_NEW_SMALL_INT(i);
+        }
     }
 
     nlr_jump(mp_obj_new_exception_msg(MP_QSTR_ValueError, "object not in sequence"));