diff options
-rw-r--r-- | alembic/testing/requirements.py | 7 | ||||
-rw-r--r-- | alembic/util.py | 1 | ||||
-rw-r--r-- | tests/test_autogen_render.py | 9 |
3 files changed, 17 insertions, 0 deletions
diff --git a/alembic/testing/requirements.py b/alembic/testing/requirements.py index a52a9b2..1e8a0a6 100644 --- a/alembic/testing/requirements.py +++ b/alembic/testing/requirements.py @@ -60,6 +60,13 @@ class SuiteRequirements(Requirements): ) @property + def fail_before_sqla_099(self): + return exclusions.fails_if( + lambda config: not util.sqla_099, + "SQLAlchemy 0.9.9 or greater required" + ) + + @property def sqlalchemy_08(self): return exclusions.skip_if( diff --git a/alembic/util.py b/alembic/util.py index 2f8e7c4..3d2c541 100644 --- a/alembic/util.py +++ b/alembic/util.py @@ -34,6 +34,7 @@ sqla_09 = _vers >= (0, 9, 0) sqla_092 = _vers >= (0, 9, 2) sqla_094 = _vers >= (0, 9, 4) sqla_094 = _vers >= (0, 9, 4) +sqla_099 = _vers >= (0, 9, 9) sqla_100 = _vers >= (1, 0, 0) if not sqla_07: raise CommandError( diff --git a/tests/test_autogen_render.py b/tests/test_autogen_render.py index 5313891..59def1b 100644 --- a/tests/test_autogen_render.py +++ b/tests/test_autogen_render.py @@ -939,6 +939,15 @@ unique=False, """ "sa.Enum('one', 'two', 'three')" ) + @config.requirements.fail_before_sqla_099 + def test_render_non_native_enum(self): + eq_ignore_whitespace( + autogenerate.render._repr_type( + Enum("one", "two", "three", native_enum=False), + self.autogen_context), + "sa.Enum('one', 'two', 'three', native_enum=False)" + ) + def test_repr_plain_sqla_type(self): type_ = Integer() autogen_context = { |