diff --git a/pytest_mock.py b/pytest_mock.py index ef4a777..6e9d011 100644 --- a/pytest_mock.py +++ b/pytest_mock.py @@ -1,12 +1,11 @@ import inspect -import sys import pytest -if sys.version_info >= (3, 3): # pragma: no cover - import unittest.mock as mock_module -else: +try: import mock as mock_module +except ImportError: + import unittest.mock as mock_module version = '1.1'