mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-20 03:20:18 +00:00
54 lines
779 B
Python
54 lines
779 B
Python
from pkpy import next
|
|
|
|
a = [1, 2, 3]
|
|
a = iter(a)
|
|
|
|
total = 0
|
|
|
|
while True:
|
|
obj = next(a)
|
|
if obj is StopIteration:
|
|
break
|
|
total += obj
|
|
|
|
assert total == 6
|
|
|
|
class Task:
|
|
def __init__(self, n):
|
|
self.n = n
|
|
|
|
def __iter__(self):
|
|
self.i = 0
|
|
return self
|
|
|
|
def __next__(self):
|
|
if self.i == self.n:
|
|
raise StopIteration
|
|
self.i += 1
|
|
return self.i
|
|
|
|
a = Task(3)
|
|
assert sum(a) == 6
|
|
|
|
i = iter(Task(5))
|
|
assert next(i) == 1
|
|
assert next(i) == 2
|
|
assert next(i) == 3
|
|
assert next(i) == 4
|
|
assert next(i) == 5
|
|
assert next(i) == StopIteration
|
|
assert next(i) == StopIteration
|
|
|
|
# test normal next
|
|
from builtins import next
|
|
|
|
a = iter([1])
|
|
assert next(a) == 1
|
|
|
|
try:
|
|
next(a)
|
|
exit(1)
|
|
except StopIteration:
|
|
pass
|
|
|