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

Fix common.iterate()

Browse Source
This commit is contained in:
Roman Zeyde
2014-08-01 10:41:47 +03:00
parent ce00e94974
commit 5508ffc0d3
3 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
common.py
View File

@@ -65,8 +65,7 @@ def iterate(data, size, func=None, truncate=True):
return
done = True
buf = np.array(buf)
result = func(buf) if func else buf
result = func(buf) if func else np.array(buf)
yield offset, result
offset += size

6
sigproc.py
View File

@@ -96,11 +96,9 @@ def coherence(x, freq):
def extract_symbols(x, freq, offset=0):
Hc = exp_iwt(-freq, Nsym) / (0.5*Nsym)
func = lambda y: np.dot(Hc, y)
iterator = common.iterate(x, Nsym, func=func)
for _, symbol in iterator:
yield symbol
for _, symbol in common.iterate(x, Nsym):
yield np.dot(Hc, symbol)
def drift(S):

2
test_common.py
View File

@@ -18,7 +18,7 @@ def test_iterate():
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) == [
assert iterlist(range(N), 1, func=lambda b: -np.array(b)) == [
(i, [-i]) for i in range(N)]