stream: use IOError when file is over.

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

View File

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

View File

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