updated and added tests

tests/test_total_annotated.py

@@ -0,0 +1,20 @@
+from pathlib import Path
+from dotenv import load_dotenv
+import os
+
+from src.database import (
+    connect,
+    total_annotated
+)
+
+if __name__ == "__main__":
+    # prepare env vars
+    env_path = Path(__file__).parent.parent / "local.env"
+    assert env_path.exists()
+    load_dotenv(env_path)
+    os.environ["MONGO_HOST"] = "localhost"
+    # connect to database
+    collection, db, client = connect()
+    # get visual communication
+    num_docs = total_annotated(collection)
+    print(f"number of annotated  documents in database: {num_docs}")