Try to import mock from unittest in AddCommandTest

9adafb91 · Jacek Sowiński · 8ebabd43 · 9adafb91
Commit 9adafb91 authored May 30, 2015 by Jacek Sowiński
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

test/AddCommandTest.py test/AddCommandTest.py +8 -1

No files found.
--- a/test/AddCommandTest.py
+++ b/test/AddCommandTest.py
@@ -16,7 +16,14 @@

 from datetime import date
 import unittest
-import mock
+
+# We're searching for 'mock'
+# pylint: disable=no-name-in-module
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
 from six import u
 from io import StringIO