diff options
Diffstat (limited to 't/unit/test_mixins.py')
| -rw-r--r-- | t/unit/test_mixins.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/t/unit/test_mixins.py b/t/unit/test_mixins.py index 372399e4..a781ca86 100644 --- a/t/unit/test_mixins.py +++ b/t/unit/test_mixins.py @@ -1,5 +1,3 @@ -from __future__ import absolute_import, unicode_literals - import pytest import socket @@ -71,7 +69,7 @@ class test_ConsumerMixin: def se2(*args, **kwargs): c.should_stop = True - raise socket.error() + raise OSError() c.connection.drain_events.side_effect = se2 it = c.consume(no_ack=True) with pytest.raises(StopIteration): |