pre-commit renamed files to PEP standard and removed unused imports

tests/generate_random_prediction_test.py

@@ -1,13 +1,15 @@
+from __future__ import annotations
+
 from pathlib import Path
 
-from src.database import VisualCommunication
+from database import VisualCommunication
 
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # get list of image paths
     test_dir = Path(__file__).parent
-    img_dir = test_dir / "imgs"
-    img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
+    img_dir = test_dir / 'imgs'
+    img_path_list = [path for path in img_dir.glob('*.jpeg') if path.is_file()]
     print(img_path_list)
     # instantiate data object
     vis_com_list = [
@@ -16,6 +18,7 @@ if __name__ == "__main__":
         in img_path_list
     ]
     # generate random predictions
-    [vis_com.generate_random_prediction() for vis_com in vis_com_list]
+    for vis_com in vis_com_list:
+        vis_com.generate_random_prediction()
     for vis_com in vis_com_list:
         print(vis_com)

tests/get_visual_communication_test.py

@@ -1,18 +1,18 @@
-from pathlib import Path
-from dotenv import load_dotenv
+from __future__ import annotations
+
 import os
+from pathlib import Path
 
-from src.database import (
-    connect,
-    get_visual_communication
-)
+from database import connect
+from database import get_visual_communication
+from dotenv import load_dotenv
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # prepare env vars
-    env_path = Path(__file__).parent.parent / "local.env"
+    env_path = Path(__file__).parent.parent / 'local.env'
     assert env_path.exists()
     load_dotenv(env_path)
-    os.environ["MONGO_HOST"] = "localhost"
+    os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
     collection, db, client = connect()
     print(client.server_info())

tests/image_download_test.py

@@ -1,15 +1,18 @@
-from pathlib import Path
-from dotenv import load_dotenv
+from __future__ import annotations
+
 import os
+from pathlib import Path
 
-from src.database import VisualCommunication, connect
+from database import connect
+from database import VisualCommunication
+from dotenv import load_dotenv
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # prepare env vars
-    env_path = Path(__file__).parent.parent / "local.env"
+    env_path = Path(__file__).parent.parent / 'local.env'
     assert env_path.exists()
     load_dotenv(env_path)
-    os.environ["MONGO_HOST"] = "localhost"
+    os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
     collection, db, client = connect()
     print(client.server_info())
@@ -17,7 +20,7 @@ if __name__ == "__main__":
     data = None
     for data in collection.find().limit(3):
         if data is None:
-            print("no document found")
+            print('no document found')
             break
         vis_com: VisualCommunication = VisualCommunication.model_validate(data)
         print(repr(vis_com))

tests/image_upload_test.py

@@ -1,15 +1,18 @@
+from __future__ import annotations
+
+import os
 from pathlib import Path
+
+from database import connect
+from database import VisualCommunication
 from dotenv import load_dotenv
 from pymongo.errors import DuplicateKeyError
-import os
 
-from src.database import VisualCommunication, connect
-
-if __name__ == "__main__":
+if __name__ == '__main__':
     # get list of image paths
     test_dir = Path(__file__).parent
-    img_dir = test_dir / "imgs"
-    img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
+    img_dir = test_dir / 'imgs'
+    img_path_list = [path for path in img_dir.glob('*.jpeg') if path.is_file()]
     print(img_path_list)
     # instantiate data object
     vis_com_list = [
@@ -20,10 +23,10 @@ if __name__ == "__main__":
     for vis_com in vis_com_list:
         print(repr(vis_com))
     # prepare env vars
-    env_path = test_dir.parent / "local.env"
+    env_path = test_dir.parent / 'local.env'
     assert env_path.exists()
     load_dotenv(env_path)
-    os.environ["MONGO_HOST"] = "localhost"
+    os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
     collection, db, client = connect()
     print(client.server_info())
@@ -32,6 +35,6 @@ if __name__ == "__main__":
         try:
             result = collection.insert_one(vis_com.model_dump())
         except DuplicateKeyError as exc:
-            print("ignoring:\n", exc)
+            print('ignoring:\n', exc)
         else:
             print(f"inserted document: {result}")

tests/model_outputs_from_annotation_test.py

@@ -1,7 +1,9 @@
-from src.database import ModelOutputs
+from __future__ import annotations
+
+from database import ModelOutputs
 
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # instantiate data object
     vis_com_list = [
         ModelOutputs.from_random()

tests/prediction_upload_test.py

@@ -1,15 +1,15 @@
-from pathlib import Path
-from dotenv import load_dotenv
-import os
+from __future__ import annotations
+
 import logging
+import os
+from pathlib import Path
 
-from src.database import (
-    VisualCommunication,
-    connect,
-    upsert_predictions
-)
+from database import connect
+from database import upsert_predictions
+from database import VisualCommunication
+from dotenv import load_dotenv
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # setup logging
     fmt = (
         '%(asctime)s | '
@@ -22,8 +22,8 @@ if __name__ == "__main__":
     logging.basicConfig(format=fmt, datefmt=datefmt, level=logging.INFO)
     # get list of image paths
     test_dir = Path(__file__).parent
-    img_dir = test_dir / "imgs"
-    img_path_list = [path for path in img_dir.glob("*.jpeg") if path.is_file()]
+    img_dir = test_dir / 'imgs'
+    img_path_list = [path for path in img_dir.glob('*.jpeg') if path.is_file()]
     # instantiate data object
     vis_com_list = [
         VisualCommunication.from_file(path)
@@ -31,16 +31,19 @@ if __name__ == "__main__":
         in img_path_list
     ]
     # generate random predictions
-    [vis_com.generate_random_prediction() for vis_com in vis_com_list]
+    for vis_com in vis_com_list:
+        vis_com.generate_random_prediction()
     # prepare env vars
-    env_path = test_dir.parent / "local.env"
+    env_path = test_dir.parent / 'local.env'
     assert env_path.exists()
     load_dotenv(env_path)
-    os.environ["MONGO_HOST"] = "localhost"
+    os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
     collection, db, client = connect()
     # upload visual communication
     for vis_com in vis_com_list:
+        if vis_com.prediction is None:
+            continue
         upsert_predictions(
             collection=collection,
             vis_com_name=vis_com.name,

tests/total_annotated_test.py

@@ -1,18 +1,18 @@
-from pathlib import Path
-from dotenv import load_dotenv
+from __future__ import annotations
+
 import os
+from pathlib import Path
 
-from src.database import (
-    connect,
-    total_annotated
-)
+from database import connect
+from database import total_annotated
+from dotenv import load_dotenv
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # prepare env vars
-    env_path = Path(__file__).parent.parent / "local.env"
+    env_path = Path(__file__).parent.parent / 'local.env'
     assert env_path.exists()
     load_dotenv(env_path)
-    os.environ["MONGO_HOST"] = "localhost"
+    os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
     collection, db, client = connect()
     # get visual communication

tests/total_documents_test.py

@@ -1,18 +1,18 @@
-from pathlib import Path
-from dotenv import load_dotenv
+from __future__ import annotations
+
 import os
+from pathlib import Path
 
-from src.database import (
-    connect,
-    total_documents
-)
+from database import connect
+from database import total_documents
+from dotenv import load_dotenv
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     # prepare env vars
-    env_path = Path(__file__).parent.parent / "local.env"
+    env_path = Path(__file__).parent.parent / 'local.env'
     assert env_path.exists()
     load_dotenv(env_path)
-    os.environ["MONGO_HOST"] = "localhost"
+    os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
     collection, db, client = connect()
     # get visual communication