diff options
-rw-r--r-- | tests/__init__.py | 2 | ||||
-rw-r--r-- | tests/test_op.py | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/tests/__init__.py b/tests/__init__.py index 9bb5ef6..cd721ae 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -29,9 +29,11 @@ testing_config.read(['test.cfg']) if py33: from unittest.mock import Mock, call + from unittest import mock else: try: from mock import Mock, call + import mock except ImportError: raise ImportError( "Alembic's test suite requires the " diff --git a/tests/test_op.py b/tests/test_op.py index 1b82af3..277f184 100644 --- a/tests/test_op.py +++ b/tests/test_op.py @@ -7,7 +7,7 @@ from sqlalchemy import event from alembic import op from . import op_fixture, assert_raises_message, requires_094, eq_ -import mock +from . import mock @event.listens_for(Table, "after_parent_attach") def _add_cols(table, metadata): |