diff --git a/pyutilib/th/__init__.py b/pyutilib/th/__init__.py index b3debdd7..2247d583 100644 --- a/pyutilib/th/__init__.py +++ b/pyutilib/th/__init__.py @@ -25,3 +25,17 @@ import pyutilib.th.nose_timeout except ImportError: pass + + +mock_available = False +try: + from unittest import mock + + mock_available = True +except ImportError: + try: + import mock + + mock_available = True + except ImportError: + pass