diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_autogen_render.py | 4 | ||||
-rw-r--r-- | tests/test_op.py | 9 | ||||
-rw-r--r-- | tests/test_postgresql.py | 3 | ||||
-rw-r--r-- | tests/test_script_production.py | 4 |
4 files changed, 9 insertions, 11 deletions
diff --git a/tests/test_autogen_render.py b/tests/test_autogen_render.py index f138df5..ada2a12 100644 --- a/tests/test_autogen_render.py +++ b/tests/test_autogen_render.py @@ -25,6 +25,7 @@ from sqlalchemy import text from sqlalchemy import types from sqlalchemy import Unicode from sqlalchemy import UniqueConstraint +from sqlalchemy import VARCHAR from sqlalchemy.engine.default import DefaultDialect from sqlalchemy.sql import and_ from sqlalchemy.sql import column @@ -1792,7 +1793,6 @@ class AutogenRenderTest(TestBase): ) def test_render_variant(self): - from sqlalchemy import VARCHAR, CHAR self.autogen_context.opts["user_module_prefix"] = None @@ -1824,8 +1824,6 @@ class AutogenRenderTest(TestBase): ) def test_repr_user_type_user_prefix_present(self): - from sqlalchemy.types import UserDefinedType - class MyType(UserDefinedType): def get_col_spec(self): return "MYTYPE" diff --git a/tests/test_op.py b/tests/test_op.py index 5682abb..c483c4a 100644 --- a/tests/test_op.py +++ b/tests/test_op.py @@ -14,10 +14,13 @@ from sqlalchemy import Table from sqlalchemy import UniqueConstraint from sqlalchemy.sql import column from sqlalchemy.sql import func +from sqlalchemy.sql import table from sqlalchemy.sql import text from sqlalchemy.sql.schema import quoted_name from alembic import op +from alembic.operations import MigrateOperation +from alembic.operations import Operations from alembic.operations import ops from alembic.operations import schemaobj from alembic.testing import assert_raises_message @@ -1023,8 +1026,6 @@ class OpTest(TestBase): def test_inline_literal(self): context = op_fixture() - from sqlalchemy.sql import table, column - from sqlalchemy import String, Integer account = table( "account", column("name", String), column("id", Integer) @@ -1144,8 +1145,6 @@ class OpTest(TestBase): class SQLModeOpTest(TestBase): def test_auto_literals(self): context = op_fixture(as_sql=True, literal_binds=True) - from sqlalchemy.sql import table, column - from sqlalchemy import String, Integer account = table( "account", column("name", String), column("id", Integer) @@ -1179,8 +1178,6 @@ class SQLModeOpTest(TestBase): class CustomOpTest(TestBase): def test_custom_op(self): - from alembic.operations import Operations, MigrateOperation - @Operations.register_operation("create_sequence") class CreateSequenceOp(MigrateOperation): """Create a SEQUENCE.""" diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py index 6a67e0b..23d6a8c 100644 --- a/tests/test_postgresql.py +++ b/tests/test_postgresql.py @@ -19,9 +19,11 @@ from sqlalchemy import types from sqlalchemy import UniqueConstraint from sqlalchemy.dialects.postgresql import ARRAY from sqlalchemy.dialects.postgresql import BYTEA +from sqlalchemy.dialects.postgresql import ExcludeConstraint from sqlalchemy.dialects.postgresql import HSTORE from sqlalchemy.dialects.postgresql import JSON from sqlalchemy.dialects.postgresql import JSONB +from sqlalchemy.dialects.postgresql import TSRANGE from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.sql import column from sqlalchemy.sql import false @@ -1248,7 +1250,6 @@ class PGUniqueIndexAutogenerateTest(AutogenFixtureTest, TestBase): @config.requirements.btree_gist def test_exclude_const_unchanged(self): - from sqlalchemy.dialects.postgresql import TSRANGE, ExcludeConstraint m1 = MetaData() m2 = MetaData() diff --git a/tests/test_script_production.py b/tests/test_script_production.py index bedf545..5c4cd9e 100644 --- a/tests/test_script_production.py +++ b/tests/test_script_production.py @@ -5,7 +5,10 @@ from unittest.mock import patch from dateutil import tz import sqlalchemy as sa +from sqlalchemy import Column from sqlalchemy import inspect +from sqlalchemy import MetaData +from sqlalchemy import Table from alembic import autogenerate from alembic import command @@ -717,7 +720,6 @@ class ImportsTest(TestBase): ) def test_imports_in_script(self): - from sqlalchemy import MetaData, Table, Column from sqlalchemy.dialects.mysql import VARCHAR type_ = VARCHAR(20, charset="utf8", national=True) |