Let mock source be chosen by Python major version.

This should provide better Python 3 compatibility, since mock is in the
Python 3 standard library as unittest.mock.

Conflicts:

	tests/test_declarations.py

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>

1 files changed, 6 insertions, 0 deletions

diff --git a/tests/compat.py b/tests/compat.py
index 8d4f1d0..769ffd4 100644
--- a/tests/compat.py
+++ b/tests/compat.py
@@ -29,3 +29,9 @@ try:
     import unittest2 as unittest
 except ImportError:
     import unittest
+
+if is_python2:
+    import mock
+else:
+    from unittest import mock
+