test command

diff --git a/tests/test_command.py b/tests/test_command.py
index b550471deb50431697813c4e24b45013e80e8841..6d70f5ad225df80fe66b4e33191531689091f783 100644 (file)
--- a/tests/test_command.py
+++ b/tests/test_command.py
@@ -1,13 +1,13 @@
-import unittest
-from . import clear_staging_env, staging_env, \
-    _sqlite_testing_config, \
-    three_rev_fixture, eq_
 from alembic import command
 from io import TextIOWrapper, BytesIO
 from alembic.script import ScriptDirectory
+from alembic.testing.fixtures import TestBase
+from alembic.testing.env import staging_env, _sqlite_testing_config, \
+    three_rev_fixture, clear_staging_env
+from alembic.testing import eq_
 
 
-class StdoutCommandTest(unittest.TestCase):
+class StdoutCommandTest(TestBase):
 
     @classmethod
     def setup_class(cls):