Implement tuple multiplication.

2 files changed, 14 insertions, 0 deletions

diff --git a/py/objtuple.c b/py/objtuple.c
index b7710e1d53..18eb6df4de 100644
--- a/py/objtuple.c
+++ b/py/objtuple.c
@@ -124,6 +124,16 @@ static mp_obj_t tuple_binary_op(int op, mp_obj_t lhs, mp_obj_t rhs) {
             m_seq_cat(s->items, o->items, o->len, p->items, p->len, mp_obj_t);
             return s;
         }
+        case RT_BINARY_OP_MULTIPLY:
+        {
+            if (!MP_OBJ_IS_SMALL_INT(rhs)) {
+                return NULL;
+            }
+            int n = MP_OBJ_SMALL_INT_VALUE(rhs);
+            mp_obj_tuple_t *s = mp_obj_new_tuple(o->len * n, NULL);
+            mp_seq_multiply(o->items, sizeof(*o->items), o->len, n, s->items);
+            return s;
+        }
         case RT_BINARY_OP_EQUAL:
         case RT_BINARY_OP_LESS:
         case RT_BINARY_OP_LESS_EQUAL:
diff --git a/tests/basics/tuple_mult.py b/tests/basics/tuple_mult.py
new file mode 100644
index 0000000000..f8350f2f27
--- /dev/null
+++ b/tests/basics/tuple_mult.py
@@ -0,0 +1,4 @@
+print((0,) * 5)
+a = (1, 2, 3)
+c = a * 3
+print(c)