dev

1 files changed, 6 insertions, 5 deletions

diff --git a/tests/test_environment.py b/tests/test_environment.py
index c877960..cb8a0a2 100644
--- a/tests/test_environment.py
+++ b/tests/test_environment.py
@@ -3,14 +3,15 @@
 from alembic.script import ScriptDirectory
 from alembic.environment import EnvironmentContext
 from alembic.migration import MigrationContext
-import unittest
-from . import Mock, call, _no_sql_testing_config, staging_env, \
-    clear_staging_env
+from alembic.testing.fixtures import TestBase
+from alembic.testing.mock import Mock, call
+from alembic.testing.env import _no_sql_testing_config, \
+    staging_env, clear_staging_env
 
-from . import eq_, is_
+from alembic.testing import eq_, is_
 
 
-class EnvironmentTest(unittest.TestCase):
+class EnvironmentTest(TestBase):
 
     def setUp(self):
         staging_env()