Add tests for __index__ function

HyeockJinKim · HyeockJinKim · commit 313242b4d6dd · 2019-09-29T22:00:33.000+09:00
Add tests for __index__ in list,
tuple, string
diff --git a/py/tests/list.py b/py/tests/list.py
@@ -111,4 +111,34 @@
 assertRaises(TypeError, lambda: list.sort(s5, key=1))
 assertRaises(TypeError, lambda: list.sort(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"
diff --git a/py/tests/string.py b/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"
diff --git a/py/tests/tuple.py b/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"
