diff options
Diffstat (limited to 'tests/test_environment.py')
| -rw-r--r-- | tests/test_environment.py | 11 |
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() |