diff --git a/tests/requirements.txt b/tests/requirements.txt index 6a91ea46..c8b4f880 100644 --- a/tests/requirements.txt +++ b/tests/requirements.txt @@ -1,5 +1,5 @@ coveralls -mock +mock; python_version < '3.3' pytest pytest-cov pytest-xdist diff --git a/tests/test_forwarder.py b/tests/test_forwarder.py index 40662d08..0e59f0b2 100644 --- a/tests/test_forwarder.py +++ b/tests/test_forwarder.py @@ -14,7 +14,10 @@ from functools import partial from contextlib import contextmanager -import mock +try: + from unittest import mock +except ImportError: + import mock import paramiko import sshtunnel import shutil