From 7c9c667633d445e4df88868d630a0af4bc63d2f8 Mon Sep 17 00:00:00 2001
From: Damien George
Date: Sat, 25 Jan 2014 00:17:36 +0000
Subject: py: Implement iterator support for object that has __getitem__.

Addresses Issue #203.
---
 tests/basics/getitem.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 tests/basics/getitem.py

(limited to 'tests/basics')

diff --git a/tests/basics/getitem.py b/tests/basics/getitem.py
new file mode 100644
index 000000000..f39296aca
--- /dev/null
+++ b/tests/basics/getitem.py
@@ -0,0 +1,22 @@
+# create a class that has a __getitem__ method
+class A:
+    def __getitem__(self, index):
+        print('getitem', index)
+        if index > 10:
+            raise StopIteration
+
+# test __getitem__
+A()[0]
+A()[1]
+
+# iterate using a for loop
+for i in A():
+    pass
+
+# iterate manually
+it = iter(A())
+try:
+    while True:
+        next(it)
+except StopIteration:
+    pass
-- 
cgit v1.2.3