updated imports

tests/total_documents_test.py

@@ -3,10 +3,11 @@ from __future__ import annotations
 import os
 from pathlib import Path
 
-from database import connect
-from database import total_documents
 from dotenv import load_dotenv
 
+from core.database import connect
+from core.database import total_documents
+
 if __name__ == '__main__':
     # prepare env vars
     env_path = Path(__file__).parent.parent / 'local.env'
@@ -16,5 +17,7 @@ if __name__ == '__main__':
     # connect to database
     collection, db, client = connect()
     # get visual communication
-    num_docs = total_documents(collection)
+    num_docs = total_documents(
+        collection=collection,
+    )
     print(f"total number of documents in database: {num_docs}")