summaryrefslogtreecommitdiff
path: root/tests/test_autogen_render.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2019-07-21 14:12:12 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2019-07-21 21:46:10 -0400
commitd27330855593fe713aa162a2a748697617bde0aa (patch)
treea7fe54b8d1d002a4e095b3759be709d7c2904c51 /tests/test_autogen_render.py
parentca053c5dde143f695ba16b14fbe4d40855459f17 (diff)
downloadalembic-d27330855593fe713aa162a2a748697617bde0aa.tar.gz
Use SQLAlchemy's testing framework fully
w/ 1.1 as the minimum version we should be able to remove the majority of the testing suite code Change-Id: I04c4ab45e2ce47f20613d6111895c6230b946f76
Diffstat (limited to 'tests/test_autogen_render.py')
-rw-r--r--tests/test_autogen_render.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_autogen_render.py b/tests/test_autogen_render.py
index 037a05f..ef2130a 100644
--- a/tests/test_autogen_render.py
+++ b/tests/test_autogen_render.py
@@ -45,9 +45,9 @@ from alembic.testing import assertions
from alembic.testing import config
from alembic.testing import eq_
from alembic.testing import eq_ignore_whitespace
+from alembic.testing import mock
from alembic.testing import TestBase
from alembic.testing.fixtures import op_fixture
-from alembic.testing.mock import patch
from alembic.util import compat
py3k = sys.version_info >= (3,)
@@ -847,7 +847,7 @@ class AutogenRenderTest(TestBase):
"SomeCustomConstraint(Column('id', Integer(), table=<t>))])",
)
- @patch("alembic.autogenerate.render.MAX_PYTHON_ARGS", 3)
+ @mock.patch("alembic.autogenerate.render.MAX_PYTHON_ARGS", 3)
def test_render_table_max_cols(self):
m = MetaData()
t = Table(