diff options
Diffstat (limited to 'tests/test_offline_environment.py')
-rw-r--r-- | tests/test_offline_environment.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/tests/test_offline_environment.py b/tests/test_offline_environment.py index 192f87f..b870dc4 100644 --- a/tests/test_offline_environment.py +++ b/tests/test_offline_environment.py @@ -1,19 +1,19 @@ -import io -from unittest import TestCase +from alembic.testing.fixtures import TestBase from alembic import command, util -from . import clear_staging_env, staging_env, \ - _no_sql_testing_config, \ - three_rev_fixture, env_file_fixture,\ - assert_raises_message + +from alembic.testing import assert_raises_message +from alembic.testing.env import staging_env, _no_sql_testing_config, \ + three_rev_fixture, clear_staging_env, env_file_fixture + a = b = c = None -class OfflineEnvironmentTest(TestCase): +class OfflineEnvironmentTest(TestBase): def setUp(self): - env = staging_env() + staging_env() self.cfg = _no_sql_testing_config() global a, b, c |