Updating model references in tests

tests/test_account.py

@@ -8,7 +8,7 @@ from pymongo.errors import ConnectionFailure
 
 from src.data.connection import ActionFailed, Collections, get_collection
 from src.data.repositories import account as account_repo
-from src.models.account import Account
+from src.models.repositories import Account
 from src.utils.logger import logger
 from tests.base_test import BaseMongoTest
 

tests/test_patient.py

@@ -6,7 +6,7 @@ from pymongo.errors import ConnectionFailure
 
 from src.data.connection import ActionFailed, Collections, get_collection
 from src.data.repositories import patient as patient_repo
-from src.models.patient import Patient
+from src.models.repositories import Patient
 from src.utils.logger import logger
 from tests.base_test import BaseMongoTest
 

tests/test_session.py

@@ -8,7 +8,7 @@ from pymongo.errors import ConnectionFailure, WriteError
 
 from src.data.connection import ActionFailed, Collections, get_collection
 from src.data.repositories import session as session_repo
-from src.models.session import Message, Session
+from src.models.repositories import Message, Session
 from src.utils.logger import logger
 from tests.base_test import BaseMongoTest