diff options
Diffstat (limited to 'tests/compat.py')
| -rw-r--r-- | tests/compat.py | 6 |
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 + |