✅ Test predict output

.gitignore

@@ -0,0 +1 @@
+__pycache__/

app.py

@@ -42,7 +42,7 @@ def metadata():
         "name": "AwesomeTox",
         "version": "1.0.0",
         "max_batch_size": 256,
-        "tox_endpoints": ["mutagenicity","hepatotoxicity"],
+        "tox_endpoints": ["NR-AR", "NR-AR-LBD", "NR-AhR", "NR-Aromatase", "NR-ER", "NR-ER-LBD", "NR-PPAR-gamma", "SR-ARE", "SR-ATAD5", "SR-HSE", "SR-MMP", "SR-p53"],
     }
 
 @app.get("/healthz")

tests/test_predict.py

@@ -1,7 +1,9 @@
 """
 Tests for the predict.py module
 """
-
+import sys
+import pathlib
+sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1]))
 from predict import predict, Tox21RandomClassifier
 
 
@@ -62,4 +64,10 @@ class TestPredictFunction:
     
     def test_predict_function_empty(self):
         result = predict([])
-        assert result == {}
+        assert result == {}
+
+#---------------------------------------------------------------------------------------
+# Debugging
+if __name__ == "__main__":
+    test = TestTox21RandomClassifier()
+    test.test_predict_multiple_smiles()