summaryrefslogtreecommitdiff
path: root/tests/__init__.py
diff options
context:
space:
mode:
authorRyan P. Kelly <rpkelly@cpan.org>2014-06-04 21:29:47 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2014-08-07 14:13:26 -0400
commitc2c28980bbb025f252402274d58bb0c6b4440e03 (patch)
treefe3cad55eb36c96fa9227f0a98541fd7e6a8a98b /tests/__init__.py
parent18722a75b3519ea53b913186cb047f60c331361a (diff)
downloadalembic-c2c28980bbb025f252402274d58bb0c6b4440e03.tar.gz
Support autogenerating migrations for tables with 255 or more columns
Diffstat (limited to 'tests/__init__.py')
-rw-r--r--tests/__init__.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/__init__.py b/tests/__init__.py
index 85ea5b3..ba8c0eb 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -29,11 +29,11 @@ testing_config = configparser.ConfigParser()
testing_config.read(['test.cfg'])
if py33:
- from unittest.mock import Mock, call
+ from unittest.mock import Mock, call, patch
from unittest import mock
else:
try:
- from mock import Mock, call
+ from mock import Mock, call, patch
import mock
except ImportError:
raise ImportError(