Merge pull request #98 from flexciton/feat/mock

Add `unittest.mock` backport
PyUtilib · Apr 27, 2020 · 10b7bb3 · 10b7bb3
2 parents 6f02a99 + ceada22
commit 10b7bb3
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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