updated and added tests

tests/test_get_visual_communication.py

@@ -0,0 +1,21 @@
+from pathlib import Path
+from dotenv import load_dotenv
+import os
+
+from src.database import (
+    connect,
+    get_visual_communication
+)
+
+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()
+    print(client.server_info())
+    # get visual communication
+    vis_com = get_visual_communication(collection)
+    print(vis_com.image)