tests/test_generate_random_prediction.py

@@ -0,0 +1,17 @@
+from pathlib import Path
+
+from src.database import VisualCommunication
+
+
+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()]
+    print(img_path_list)
+    # instantiate data object
+    vis_com_list = [VisualCommunication.from_file(path) for path 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:
+        print(vis_com)