lili/amodem
1
0
Fork 0
mirror of https://github.com/romanz/amodem.git synced 2026-03-17 07:05:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

simplify common.iterate()

Browse Source
This commit is contained in:
Roman Zeyde
2014-07-12 16:18:06 +03:00
parent cb28d33908
commit b261f1b102
4 changed files with 31 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
test_common.py
View File

@@ -1,20 +1,26 @@
import common
import numpy as np
def iterlist(x, *args, **kwargs):
x = np.array(x)
return [(offset, list(buf)) for offset, buf in common.iterate(x, *args, **kwargs)]
return list((i, list(x)) for i, x in common.iterate(x, *args, **kwargs))
def test_iterate():
N = 10
assert iterlist(range(N), 1) == [(i, [i]) for i in range(N)]
assert iterlist(range(N), 1) == [(i, [i]) for i in range(N)]
assert iterlist(range(N), 2) == [(i, [i, i+1]) for i in range(N-1)]
assert iterlist(range(N), 3) == [(i, [i, i+1, i+2]) for i in range(N-2)]
assert iterlist(range(N), 3, advance=2) == [(i, [i, i+1, i+2]) for i in range(0, N-2, 2)]
assert iterlist(range(N), 3, advance=3) == [(i, [i, i+1, i+2]) for i in range(0, N-2, 3)]
assert iterlist(range(N), 2, offset=5) == [(i, [i, i+1]) for i in range(5, N-1)]
assert iterlist(range(N), 1, func=lambda b: -b) == [(i, [-i]) for i in range(N)]
assert iterlist(range(N), 1) == [
(i, [i]) for i in range(N)]
assert iterlist(range(N), 2) == [
(i, [i, i+1]) for i in range(0, N-1, 2)]
assert iterlist(range(N), 3) == [
(i, [i, i+1, i+2]) for i in range(0, N-2, 3)]
assert iterlist(range(N), 1, func=lambda b: -b) == [
(i, [-i]) for i in range(N)]
def test_split():
L = [(i*2, i*2+1) for i in range(10)]
@@ -30,6 +36,7 @@ def test_split():
except IndexError as e:
assert e.args == (i,)
def test_icapture():
x = range(100)
y = []