dev

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