diff options
Diffstat (limited to 'tests/test_command_helpers.py')
-rw-r--r-- | tests/test_command_helpers.py | 6 |
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 |