python-dockercloud/tests/__init__.py
2016-01-26 18:45:04 +01:00

12 lines
391 B
Python

__author__ = 'fermayo'
# Python 3.4.2 includes mock in-box, prefer that version
# For other versions, patch it up to use the external mock library
try:
from unittest import mock
except ImportError:
import sys
sys.modules['unittest'] = __import__('unittest')
sys.modules['unittest.mock'] = __import__('mock')
setattr(sys.modules['unittest'], 'mock', __import__('mock'))