import mock package correctly under 2.7

Author: will-ockmore

tests/test_cli.py

@@ -4,7 +4,11 @@
 
 import json
 import os
-from unittest.mock import MagicMock
+
+try:
+    from unittest.mock import MagicMock
+except ImportError:
+    from mock import MagicMock
 
 import pytest