From 1dbc585f9ebf35bba63dc127e2ea3b565b9dd9bb Mon Sep 17 00:00:00 2001
From: Patience Shyu <patienceshyu@gmail.com>
Date: Tue, 31 Jul 2018 16:04:53 +0200
Subject: [PATCH] add __setitem__ methods and tests

---
 python/common/org/python/types/Bool.java      | 10 ++++++++++
 python/common/org/python/types/FrozenSet.java | 10 ++++++++++
 python/common/org/python/types/Int.java       | 10 ++++++++++
 python/common/org/python/types/Range.java     | 10 ++++++++++
 python/common/org/python/types/Str.java       | 10 ++++++++++
 tests/datatypes/test_bool.py                  |  9 +++++++++
 tests/datatypes/test_float.py                 |  9 +++++++++
 tests/datatypes/test_frozenset.py             |  9 +++++++++
 tests/datatypes/test_int.py                   |  9 +++++++++
 tests/datatypes/test_range.py                 |  9 +++++++++
 tests/datatypes/test_str.py                   | 10 ++++++++++
 tests/datatypes/test_tuple.py                 |  9 +++++++++
 12 files changed, 114 insertions(+)

diff --git a/python/common/org/python/types/Bool.java b/python/common/org/python/types/Bool.java
index 1b3f801fed..18935c7d95 100644
--- a/python/common/org/python/types/Bool.java
+++ b/python/common/org/python/types/Bool.java
@@ -94,6 +94,16 @@ public org.python.types.Str __getitem__(org.python.Object format_string) {
         throw new org.python.exceptions.TypeError("'bool' object is not subscriptable");
     }
 
+    @org.python.Method(
+            __doc__ = "",
+            args = {"index", "value"}
+    )
+    public void __setitem__(org.python.Object index, org.python.Object value) {
+        throw new org.python.exceptions.TypeError(
+                "'bool' object does not support item assignment"
+        );
+    }
+
     @org.python.Method(
             __doc__ = "Return self<value.",
             args = {"other"}
diff --git a/python/common/org/python/types/FrozenSet.java b/python/common/org/python/types/FrozenSet.java
index dafe2de2b5..0583b8786f 100644
--- a/python/common/org/python/types/FrozenSet.java
+++ b/python/common/org/python/types/FrozenSet.java
@@ -165,6 +165,16 @@ public org.python.Object copy() {
         return this;
     }
 
+    @org.python.Method(
+            __doc__ = "",
+            args = {"index", "value"}
+    )
+    public void __setitem__(org.python.Object index, org.python.Object value) {
+        throw new org.python.exceptions.TypeError(
+                "'frozenset' object does not support item assignment"
+        );
+    }
+
     @org.python.Method(
             __doc__ = "Return self<value.",
             args = {"other"}
diff --git a/python/common/org/python/types/Int.java b/python/common/org/python/types/Int.java
index d13a19fca3..0bad6dac61 100644
--- a/python/common/org/python/types/Int.java
+++ b/python/common/org/python/types/Int.java
@@ -155,6 +155,16 @@ public org.python.types.Str __getitem__(org.python.Object format_str) {
         throw new org.python.exceptions.TypeError("'int' object is not subscriptable");
     }
 
+    @org.python.Method(
+            __doc__ = "",
+            args = {"index", "value"}
+    )
+    public void __setitem__(org.python.Object index, org.python.Object value) {
+        throw new org.python.exceptions.TypeError(
+                "'int' object does not support item assignment"
+        );
+    }
+
     @org.python.Method(
             __doc__ = "Return self<value.",
             args = {"other"}
diff --git a/python/common/org/python/types/Range.java b/python/common/org/python/types/Range.java
index a569de3db8..540da317ee 100644
--- a/python/common/org/python/types/Range.java
+++ b/python/common/org/python/types/Range.java
@@ -122,6 +122,16 @@ public org.python.Object __getitem__(org.python.Object index) {
         }
     }
 
+    @org.python.Method(
+            __doc__ = "",
+            args = {"index", "value"}
+    )
+    public void __setitem__(org.python.Object index, org.python.Object value) {
+        throw new org.python.exceptions.TypeError(
+                "'range' object does not support item assignment"
+        );
+    }
+
     @org.python.Method(
             __doc__ = "Implement __len__(self)."
     )
diff --git a/python/common/org/python/types/Str.java b/python/common/org/python/types/Str.java
index 6b4f30160e..8a629e06fe 100644
--- a/python/common/org/python/types/Str.java
+++ b/python/common/org/python/types/Str.java
@@ -376,6 +376,16 @@ public org.python.Object __getitem__(org.python.Object index) {
         }
     }
 
+    @org.python.Method(
+            __doc__ = "",
+            args = {"index", "value"}
+    )
+    public void __setitem__(org.python.Object index, org.python.Object value) {
+        throw new org.python.exceptions.TypeError(
+                "'str' object does not support item assignment"
+        );
+    }
+
     @org.python.Method(
             __doc__ = "Implement iter(self)."
     )
diff --git a/tests/datatypes/test_bool.py b/tests/datatypes/test_bool.py
index 60118df098..2058d42d64 100644
--- a/tests/datatypes/test_bool.py
+++ b/tests/datatypes/test_bool.py
@@ -21,6 +21,15 @@ def test_getattr(self):
                 print(err)
             """)
 
+    def test_setitem(self):
+        self.assertCodeExecution("""
+            x = True
+            try:
+                x[0] = 1
+            except TypeError as err:
+                print(err)
+        """)
+
 
 class UnaryBoolOperationTests(UnaryOperationTestCase, TranspileTestCase):
     data_type = 'bool'
diff --git a/tests/datatypes/test_float.py b/tests/datatypes/test_float.py
index 124ea38e47..06214b8aaa 100644
--- a/tests/datatypes/test_float.py
+++ b/tests/datatypes/test_float.py
@@ -26,6 +26,15 @@ def test_getattr(self):
                 print(err)
             """)
 
+    def test_setitem(self):
+        self.assertCodeExecution("""
+            x = 3.14159
+            try:
+                x[0] = 2
+            except TypeError as err:
+                print(err)
+        """)
+
     def test_repr(self):
         self.assertCodeExecution("""
             x = 350000000000000000.0
diff --git a/tests/datatypes/test_frozenset.py b/tests/datatypes/test_frozenset.py
index feee2b1114..34442af86f 100644
--- a/tests/datatypes/test_frozenset.py
+++ b/tests/datatypes/test_frozenset.py
@@ -145,6 +145,15 @@ def test_copy(self):
 
             """)
 
+    def test_setitem(self):
+        self.assertCodeExecution("""
+            x = frozenset("hello world")
+            try:
+                x[0] = "goodbye"
+            except TypeError as err:
+                print(err)
+        """)
+
     def test_isdisjoint(self):
         self.assertCodeExecution("""
             x = frozenset("hello world")
diff --git a/tests/datatypes/test_int.py b/tests/datatypes/test_int.py
index edbd854927..cec0deb644 100644
--- a/tests/datatypes/test_int.py
+++ b/tests/datatypes/test_int.py
@@ -28,6 +28,15 @@ def test_getattr(self):
                 print(err)
             """)
 
+    def test_setitem(self):
+        self.assertCodeExecution("""
+            x = 37
+            try:
+                x[0] = 1
+            except TypeError as err:
+                print(err)
+        """)
+
     def test_invalid_literal(self):
         self.assertCodeExecution("""
             try:
diff --git a/tests/datatypes/test_range.py b/tests/datatypes/test_range.py
index 152746e924..05827e838b 100644
--- a/tests/datatypes/test_range.py
+++ b/tests/datatypes/test_range.py
@@ -80,6 +80,15 @@ def test_iterator_iterator(self):
             print(r)
         """)
 
+    def test_setitem(self):
+        self.assertCodeExecution("""
+            r = range(10)
+            try:
+                r[0] = "abc"
+            except TypeError as e:
+                print(e)
+        """)
+
 
 class UnaryRangeOperationTests(UnaryOperationTestCase, TranspileTestCase):
     data_type = 'range'
diff --git a/tests/datatypes/test_str.py b/tests/datatypes/test_str.py
index 2869e81aaa..51b3fa02aa 100644
--- a/tests/datatypes/test_str.py
+++ b/tests/datatypes/test_str.py
@@ -238,6 +238,16 @@ def test_getitem(self):
                 print(err)
             """)
 
+    def test_setitem(self):
+        # Strings are immutable and do not allow item assignment
+        self.assertCodeExecution("""
+            x = "BeeWare"
+            try:
+                x[0] = "A"
+            except TypeError as err:
+                print(err)
+        """)
+
     def test_slice(self):
         # Full slice
         self.assertCodeExecution("""
diff --git a/tests/datatypes/test_tuple.py b/tests/datatypes/test_tuple.py
index bebe0df4cc..6f56e23a19 100644
--- a/tests/datatypes/test_tuple.py
+++ b/tests/datatypes/test_tuple.py
@@ -43,6 +43,15 @@ def test_const_creation_multitype(self):
             print(x)
             """)
 
+    def test_setitem(self):
+        self.assertCodeExecution("""
+            x = (1, 2, 3, 4, 5)
+            try:
+                x[0] = "a"
+            except TypeError as err:
+                print(err)
+        """)
+
     def test_getitem(self):
         # Simple positive index
         self.assertCodeExecution("""