summaryrefslogtreecommitdiff
path: root/tests/test_command_helpers.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_command_helpers.py')
-rw-r--r--tests/test_command_helpers.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/test_command_helpers.py b/tests/test_command_helpers.py
index 140e915..af81601 100644
--- a/tests/test_command_helpers.py
+++ b/tests/test_command_helpers.py
@@ -13,8 +13,12 @@
# See the License for the specific language governing permissions and
# limitations under the License.
+try:
+ from unittest import mock
+except ImportError:
+ import mock
+
from six import StringIO
-import mock
import testtools
from swiftclient import command_helpers as h