diff options
Diffstat (limited to 'tests/test_virtualenv.py')
| -rw-r--r-- | tests/test_virtualenv.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/test_virtualenv.py b/tests/test_virtualenv.py index 27f3d10..a48c5e4 100644 --- a/tests/test_virtualenv.py +++ b/tests/test_virtualenv.py @@ -5,10 +5,14 @@ import sys import tempfile import pytest -from mock import NonCallableMock, call, patch import virtualenv +try: + from unittest.mock import NonCallableMock, call, patch +except ImportError: + from mock import NonCallableMock, call, patch + def test_version(): """Should have a version string""" |