diff options
Diffstat (limited to 'tests/tools.py')
| -rw-r--r-- | tests/tools.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/tools.py b/tests/tools.py index 5c65847..73a64b6 100644 --- a/tests/tools.py +++ b/tests/tools.py @@ -37,9 +37,9 @@ from azurelinuxagent.common.version import PY_VERSION_MAJOR # Import mock module for Python2 and Python3 try: - from unittest.mock import Mock, patch, MagicMock, DEFAULT, ANY, call + from unittest.mock import Mock, patch, MagicMock, DEFAULT, call except ImportError: - from mock import Mock, patch, MagicMock, DEFAULT, ANY, call + from mock import Mock, patch, MagicMock, DEFAULT, call test_dir = os.path.dirname(os.path.abspath(__file__)) data_dir = os.path.join(test_dir, "data") |