mirror of
https://github.com/romanz/amodem.git
synced 2026-02-07 18:08:03 +08:00
19 lines
845 B
Python
19 lines
845 B
Python
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)]
|
|
|
|
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)]
|
|
|