add try/except for the tests related to unittest.mock

cokelaer · Feb 20, 2017 · 03de436 · 03de436
1 parent 664ee27
commit 03de436
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/test/test_browse.py b/test/test_browse.py
@@ -1,5 +1,9 @@
 from easydev import browser
-from unittest.mock import patch
+
+try:
+    from unittest.mock import patch
+except:
+    from mock import patch
 
 
 def test_browse(mocker):

diff --git a/test/test_platform.py b/test/test_platform.py
@@ -1,7 +1,10 @@
 from easydev.platform import get_platform, linux_distribution, is_windows
 from easydev.platform import is_linux, is_mac
 
-from unittest.mock import patch
+try:
+    from unittest.mock import patch
+except:
+    from mock import patch