Add list addition (fixes: #39)

2 files changed, 21 insertions, 0 deletions

diff --git a/py/objlist.c b/py/objlist.c
index ce16ab6fe9..b488a1964e 100644
--- a/py/objlist.c
+++ b/py/objlist.c
@@ -18,6 +18,7 @@ typedef struct _mp_obj_list_t {
 } mp_obj_list_t;
 
 static mp_obj_t mp_obj_new_list_iterator(mp_obj_list_t *list, int cur);
+static mp_obj_list_t *list_new(uint n);
 
 /******************************************************************************/
 /* list                                                                       */
@@ -43,6 +44,21 @@ static mp_obj_t list_binary_op(int op, mp_obj_t lhs, mp_obj_t rhs) {
             uint index = mp_get_index(o->base.type, o->len, rhs);
             return o->items[index];
         }
+        case RT_BINARY_OP_ADD:
+        {
+	    if (!MP_OBJ_IS_TYPE(rhs, &list_type)) {
+		return NULL;
+	    }
+	    mp_obj_list_t *p = rhs;
+	    mp_obj_list_t *s = list_new(o->len + p->len);
+	    for (int i = 0; i < o->len; i++) {
+		s->items[i] = o->items[i];
+	    }
+	    for (int i = 0; i < p->len; i++) {
+		s->items[i+o->len] = p->items[i];
+	    }
+	    return s;
+        }
         default:
             // op not supported
             return NULL;
diff --git a/tests/basics/tests/list2.py b/tests/basics/tests/list2.py
new file mode 100644
index 0000000000..e46042b86a
--- /dev/null
+++ b/tests/basics/tests/list2.py
@@ -0,0 +1,5 @@
+# list addition
+a = [1,2,3]
+b = [4,5,6]
+c = a + b
+print(c)