py: Implement +, += and .extend for bytearray and array objs.

Addresses issue #994.

2 files changed, 26 insertions, 0 deletions

diff --git a/tests/basics/array_add.py b/tests/basics/array_add.py
new file mode 100644
index 000000000..1dba8a3bc
--- /dev/null
+++ b/tests/basics/array_add.py
@@ -0,0 +1,12 @@
+# test array + array
+import array
+
+a1 = array.array('I', [1])
+a2 = array.array('I', [2])
+print(a1 + a2)
+
+a1 += array.array('I', [3, 4])
+print(a1)
+
+a1.extend(array.array('I', [5]))
+print(a1)
diff --git a/tests/basics/bytearray_add.py b/tests/basics/bytearray_add.py
new file mode 100644
index 000000000..21a386c6e
--- /dev/null
+++ b/tests/basics/bytearray_add.py
@@ -0,0 +1,14 @@
+# test bytearray + bytearray
+
+b = bytearray(2)
+b[0] = 1
+b[1] = 2
+print(b + bytearray(2))
+
+# inplace add
+b += bytearray(3)
+print(b)
+
+# extend
+b.extend(bytearray(4))
+print(b)