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

main: refactor send and recv

Browse Source
This commit is contained in:
Roman Zeyde
2015-02-04 08:16:12 +02:00
parent 53559ff8df
commit ef8a75f10e
5 changed files with 76 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_transfer.py
View File

@@ -3,8 +3,7 @@ from io import BytesIO
import numpy as np
from amodem import send
from amodem import recv
from amodem import main
from amodem import common
from amodem import dsp
from amodem import sampling
@@ -30,7 +29,7 @@ class Args(object):
def run(size, chan=None, df=0, success=True, reader=None):
tx_data = os.urandom(size)
tx_audio = BytesIO()
send.main(config=config, src=BytesIO(tx_data), dst=tx_audio)
main.send(config=config, src=BytesIO(tx_data), dst=tx_audio)
data = tx_audio.getvalue()
data = common.loads(data)
@@ -49,7 +48,7 @@ def run(size, chan=None, df=0, success=True, reader=None):
if reader:
rx_audio = reader(rx_audio)
try:
result = recv.main(config=config, src=rx_audio, dst=rx_data,
result = main.recv(config=config, src=rx_audio, dst=rx_data,
dump_audio=dump)
finally:
rx_audio.close()