Merge pull request #651 from kostya/insert

Array#insert
topazproject · Apr 21, 2013 · dbce6f2 · dbce6f2
2 parents 6ffe810 + 64badca
commit dbce6f2
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 12 deletions.
diff --git a/spec/tags/core/array/insert_tags.txt b/spec/tags/core/array/insert_tags.txt
diff --git a/topaz/objects/arrayobject.py b/topaz/objects/arrayobject.py
@@ -323,3 +323,26 @@ def method_rotate_i(self, space, n=1):
         self.items_w.extend(self.items_w[:n])
         del self.items_w[:n]
         return self
+
+    @classdef.method("insert", i="int")
+    @check_frozen()
+    def method_insert(self, space, i, args_w):
+        if not args_w:
+            return self
+        length = len(self.items_w)
+        if i > length:
+            for _ in xrange(i - length):
+                self.items_w.append(space.w_nil)
+            self.items_w.extend(args_w)
+            return self
+        if i < 0:
+            if i < -length - 1:
+                 raise space.error(space.w_IndexError,
+                      "index %d too small for array; minimum: %d" % (i + 1, -length))
+            i += length + 1
+        assert i >= 0
+        j = i
+        for e in args_w:
+            self.items_w.insert(j, e)
+            j += 1
+        return self