fix iterator issues

This commit is contained in:
Roman Zeyde
2014-08-07 17:03:09 +03:00
parent 16bc186631
commit 53363ee171
6 changed files with 12 additions and 10 deletions

View File

@@ -25,13 +25,13 @@ def test_iterate():
def test_split():
L = [(i*2, i*2+1) for i in range(10)]
iters = common.split(L, n=2)
assert zip(*iters) == L
assert list(zip(*iters)) == L
for i in [0, 1]:
iters = common.split(L, n=2)
iters[i].next()
next(iters[i])
try:
iters[i].next()
next(iters[i])
assert False
except IndexError as e:
assert e.args == (i,)
@@ -43,8 +43,8 @@ def test_icapture():
z = []
for i in common.icapture(x, result=y):
z.append(i)
assert x == y
assert x == z
assert list(x) == y
assert list(x) == z
def test_dumps_loads():