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

stream: use IOError when file is over.

Browse Source
This commit is contained in:
Roman Zeyde
2014-09-06 08:27:53 +03:00
parent 2f352bcbe2
commit 9cdabd938a
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
amodem/stream.py
View File

@@ -4,10 +4,6 @@ import logging
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
class Timeout(Exception):
pass
class Reader(object): class Reader(object):
def __init__(self, fd, data_type=None, bufsize=4096, def __init__(self, fd, data_type=None, bufsize=4096,
@@ -51,6 +47,6 @@ class Reader(object):
time.sleep(self.wait) time.sleep(self.wait)
raise Timeout(self.timeout) raise IOError('timeout')
__next__ = next __next__ = next

4
tests/test_stream.py
View File

@@ -31,5 +31,5 @@ def test_read():
try: try:
for buf in f: for buf in f:
pass pass
except stream.Timeout as e: except IOError as e:
assert e.args == (f.timeout,) assert e.args == ('timeout',)