blob: 763f59cb6fd01a1a6a1ca3f9fac9a5bb83b7b379 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
# user defined iterator used in something other than a for loop
class MyStopIteration(StopIteration):
pass
class myiter:
def __init__(self, i):
self.i = i
def __iter__(self):
return self
def __next__(self):
if self.i == 0:
raise StopIteration
elif self.i == 1:
raise StopIteration(1)
elif self.i == 2:
raise MyStopIteration
print(list(myiter(0)))
print(list(myiter(1)))
print(list(myiter(2)))
|