From 47cbc8ba6c98b44ff438d37efcd06c54101d61aa Mon Sep 17 00:00:00 2001 From: Jonas Scholl Date: Sat, 10 Dec 2022 22:47:53 +0100 Subject: [PATCH] remove imports from root module --- database_setup_tools/setup.py | 2 +- tests/integration/test_database_setup.py | 2 +- tests/unit/test_session_manager.py | 2 +- tests/unit/test_setup.py | 3 ++- 4 files changed, 5 insertions(+), 4 deletions(-) diff --git a/database_setup_tools/setup.py b/database_setup_tools/setup.py index efe371c..e6e4613 100644 --- a/database_setup_tools/setup.py +++ b/database_setup_tools/setup.py @@ -3,7 +3,7 @@ import sqlalchemy_utils from sqlalchemy import MetaData -from database_setup_tools import SessionManager +from database_setup_tools.session_manager import SessionManager class DatabaseSetup: diff --git a/tests/integration/test_database_setup.py b/tests/integration/test_database_setup.py index 4984f4f..8c0d788 100644 --- a/tests/integration/test_database_setup.py +++ b/tests/integration/test_database_setup.py @@ -1,7 +1,7 @@ import pytest from starlette.testclient import TestClient -from database_setup_tools import DatabaseSetup +from database_setup_tools.setup import DatabaseSetup from tests.integration.example.app import app, model_metadata, DATABASE_URI database_setup = DatabaseSetup(model_metadata=model_metadata, database_uri=DATABASE_URI) diff --git a/tests/unit/test_session_manager.py b/tests/unit/test_session_manager.py index fced000..88616b0 100644 --- a/tests/unit/test_session_manager.py +++ b/tests/unit/test_session_manager.py @@ -1,6 +1,6 @@ import pytest as pytest -from database_setup_tools import SessionManager +from database_setup_tools.session_manager import SessionManager class TestSessionManager: diff --git a/tests/unit/test_setup.py b/tests/unit/test_setup.py index 9ccf614..38cd530 100644 --- a/tests/unit/test_setup.py +++ b/tests/unit/test_setup.py @@ -2,7 +2,8 @@ from sqlalchemy.exc import OperationalError from sqlalchemy.orm.scoping import ScopedSession -from database_setup_tools import DatabaseSetup, SessionManager +from database_setup_tools.session_manager import SessionManager +from database_setup_tools.setup import DatabaseSetup from tests.unit.sample_model import model_metadata