lili/amodem
1
0
Fork 0
mirror of https://github.com/romanz/amodem.git synced 2026-04-20 21:26:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add noisy channel support

Browse Source
This commit is contained in:
Roman Zeyde
2014-08-04 18:20:02 +03:00
parent 6c6c090e79
commit d94675e7d7
1 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
amodem/test/test_full.py
View File

@@ -1,15 +1,20 @@
import os import os
from cStringIO import StringIO from cStringIO import StringIO
import numpy as np
import send import send
import recv import recv
import common import common
import logging
logging.basicConfig(level=logging.DEBUG,
format='%(asctime)s %(levelname)-12s %(message)s')
class Args(object): class Args(object):
def __init__(self, **kwargs): def __init__(self, **kwargs):
self.__dict__.update(kwargs) self.__dict__.update(kwargs)
def run(chan, size): def run(size, chan):
tx_data = os.urandom(size) tx_data = os.urandom(size)
tx_audio = StringIO() tx_audio = StringIO()
send.main(Args(silence_start=1, silence_stop=1, input=StringIO(tx_data), output=tx_audio)) send.main(Args(silence_start=1, silence_stop=1, input=StringIO(tx_data), output=tx_audio))
@@ -27,7 +32,18 @@ def run(chan, size):
assert rx_data == tx_data assert rx_data == tx_data
def test_small(): def test_small():
run(chan=lambda x: x, size=1024) run(1024, lambda x: x)
def test_large(): def test_large():
run(chan=lambda x: x, size=64 << 10) run(54321, lambda x: x)
def test_attenuation():
run(5120, lambda x: x * 0.1)
def test_low_noise():
r = np.random.RandomState(seed=0)
run(5120, lambda x: x + r.normal(size=len(x), scale=0.0001))
def test_medium_noise():
r = np.random.RandomState(seed=0)
run(5120, lambda x: x + r.normal(size=len(x), scale=0.001))