diff options
Diffstat (limited to 'tests/test_retry.py')
| -rw-r--r-- | tests/test_retry.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/test_retry.py b/tests/test_retry.py index 535485a..c4650bc 100644 --- a/tests/test_retry.py +++ b/tests/test_retry.py @@ -1,8 +1,8 @@ -from redis.backoff import NoBackoff import pytest -from redis.exceptions import ConnectionError +from redis.backoff import NoBackoff from redis.connection import Connection, UnixDomainSocketConnection +from redis.exceptions import ConnectionError from redis.retry import Retry @@ -34,8 +34,7 @@ class TestConnectionConstructorWithRetry: @pytest.mark.parametrize("Class", [Connection, UnixDomainSocketConnection]) def test_retry_on_timeout_retry(self, Class, retries): retry_on_timeout = retries > 0 - c = Class(retry_on_timeout=retry_on_timeout, - retry=Retry(NoBackoff(), retries)) + c = Class(retry_on_timeout=retry_on_timeout, retry=Retry(NoBackoff(), retries)) assert c.retry_on_timeout == retry_on_timeout assert isinstance(c.retry, Retry) assert c.retry._retries == retries |