remove imports from root module

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:

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)

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:

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