main: use amodem package for recv.main() and send.main() API

This commit is contained in:
Roman Zeyde
2014-12-28 16:46:47 +02:00
parent 0ae80e6d8b
commit fbd34844cf
5 changed files with 28 additions and 35 deletions

View File

@@ -29,8 +29,7 @@ class Args(object):
def run(size, chan=None, df=0, success=True):
tx_data = os.urandom(size)
tx_audio = BytesIO()
send.main(Args(config=config, silence_start=1, silence_stop=1,
input=BytesIO(tx_data), output=tx_audio))
send.main(config=config, src=BytesIO(tx_data), dst=tx_audio)
data = tx_audio.getvalue()
data = common.loads(data)
@@ -45,8 +44,7 @@ def run(size, chan=None, df=0, success=True):
rx_audio = BytesIO(data)
rx_data = BytesIO()
result = recv.main(Args(config=config,
skip=0, input=rx_audio, output=rx_data))
result = recv.main(config=config, src=rx_audio, dst=rx_data)
rx_data = rx_data.getvalue()
assert result == success