From 5d4a8213395b1e27d303379772c90b3b0adc82d5 Mon Sep 17 00:00:00 2001
From: "John R. Lenton" <jlenton@gmail.com>
Date: Sat, 4 Jan 2014 00:26:30 +0000
Subject: [PATCH] Implements list.index. Fixes issue #57.

---
 py/objlist.c                     | 22 ++++++++++++++++++++++
 tests/basics/tests/list_index.py | 12 ++++++++++++
 2 files changed, 34 insertions(+)
 create mode 100644 tests/basics/tests/list_index.py

diff --git a/py/objlist.c b/py/objlist.c
index 8b841d4bf0..20409a26ce 100644
--- a/py/objlist.c
+++ b/py/objlist.c
@@ -153,10 +153,31 @@ static mp_obj_t list_count(mp_obj_t self_in, mp_obj_t value) {
     return mp_obj_new_int(count);
 }
 
+static mp_obj_t list_index(int n_args, const mp_obj_t *args) {
+    assert(2 <= n_args && n_args <= 4);
+    assert(MP_OBJ_IS_TYPE(args[0], &list_type));
+    mp_obj_list_t *self = args[0];
+    mp_obj_t *value = args[1];
+
+    uint start = mp_get_index(self->base.type, self->len,
+                              n_args >= 3 ? args[2] : mp_obj_new_int(0));
+    uint stop = mp_get_index(self->base.type, self->len,
+                             n_args >= 4 ? args[3] : mp_obj_new_int(-1));
+
+    for (uint i = start; i <= stop; i++) {
+         if (mp_obj_equal(self->items[i], value)) {
+              return mp_obj_new_int(i);
+         }
+    }
+
+    nlr_jump(mp_obj_new_exception_msg(rt_q_ValueError, "Object not in list."));
+}
+
 static MP_DEFINE_CONST_FUN_OBJ_2(list_append_obj, mp_obj_list_append);
 static MP_DEFINE_CONST_FUN_OBJ_1(list_clear_obj, list_clear);
 static MP_DEFINE_CONST_FUN_OBJ_1(list_copy_obj, list_copy);
 static MP_DEFINE_CONST_FUN_OBJ_2(list_count_obj, list_count);
+static MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(list_index_obj, 2, 4, list_index);
 static MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(list_pop_obj, 1, 2, list_pop);
 static MP_DEFINE_CONST_FUN_OBJ_2(list_sort_obj, list_sort);
 
@@ -174,6 +195,7 @@ const mp_obj_type_t list_type = {
         { "clear", &list_clear_obj },
         { "copy", &list_copy_obj },
         { "count", &list_count_obj },
+        { "index", &list_index_obj },
         { "pop", &list_pop_obj },
         { "sort", &list_sort_obj },
         { NULL, NULL }, // end-of-list sentinel
diff --git a/tests/basics/tests/list_index.py b/tests/basics/tests/list_index.py
new file mode 100644
index 0000000000..cc853fe0e7
--- /dev/null
+++ b/tests/basics/tests/list_index.py
@@ -0,0 +1,12 @@
+a = [1, 2, 3]
+a = a + a
+b = [0, 0, a]
+print(a.index(2))
+print(b.index(a))
+print(a.index(2, 2))
+try:
+    a.index(2, 2, 2)
+except ValueError:
+    print("Raised ValueError")
+else:
+    raise AssertionError("Did not raise ValueError")