Add tests for __index__

Add tests for __index__ function
in list, string, tuple

Author: HyeockJinKim

py/tests/list.py

@@ -39,4 +39,34 @@
 assert a * 0 == []
 assert a * -1 == []
 
+class Index:
+    def __index__(self):
+        return 1
+
+a = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+b = Index()
+assert a[b] == 1
+assert a[b:10] == a[1:10]
+assert a[10:b:-1] == a[10:1:-1]
+
+class NonIntegerIndex:
+    def __index__(self):
+        return 1.1
+
+a = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+b = NonIntegerIndex()
+try:
+    a[b]
+except TypeError:
+    pass
+else:
+    assert False, "TypeError not raised"
+
+try:
+    a[b:10]
+except TypeError:
+    pass
+else:
+    assert False, "TypeError not raised"
+
 doc="finished"

py/tests/string.py

@@ -886,4 +886,35 @@ def index(s, i):
 assert uni[7:7:2] == ''
 assert uni[7:7:3] == ''
 
+class Index:
+    def __index__(self):
+        return 1
+
+a = '012345678910'
+b = Index()
+assert a[b] == '1'
+assert a[b:10] == a[1:10]
+assert a[10:b:-1] == a[10:1:-1]
+
+class NonIntegerIndex:
+    def __index__(self):
+        return 1.1
+
+a = '012345678910'
+b = NonIntegerIndex()
+try:
+    a[b]
+except TypeError:
+    pass
+else:
+    assert False, "TypeError not raised"
+
+try:
+    a[b:10]
+except TypeError:
+    pass
+else:
+    assert False, "TypeError not raised"
+
+
 doc="finished"

py/tests/tuple.py

@@ -20,4 +20,34 @@
 assert a * 0 == ()
 assert a * -1 == ()
 
+class Index:
+    def __index__(self):
+        return 1
+
+a = (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
+b = Index()
+assert a[b] == 1
+assert a[b:10] == a[1:10]
+assert a[10:b:-1] == a[10:1:-1]
+
+class NonIntegerIndex:
+    def __index__(self):
+        return 1.1
+
+a = (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
+b = NonIntegerIndex()
+try:
+    a[b]
+except TypeError:
+    pass
+else:
+    assert False, "TypeError not raised"
+
+try:
+    a[b:10]
+except TypeError:
+    pass
+else:
+    assert False, "TypeError not raised"
+
 doc="finished"