tests/basics/builtin_next_arg2.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

# test next(iter, default)

try:
    next(iter([]), 42)
except TypeError:  # 2-argument version not supported
    print('SKIP')
    raise SystemExit

print(next(iter([]), 42))
print(next(iter(range(0)), 42))
print(next((x for x in [0] if x == 1), 43))

def gen():
    yield 1
    yield 2

g = gen()
print(next(g, 42))
print(next(g, 43))
print(next(g, 44))

class Gen:
    def __init__(self):
        self.b = False

    def __next__(self):
        if self.b:
            raise StopIteration
        self.b = True
        return self.b

g = Gen()
print(next(g, 44))
print(next(g, 45))