Init test for submitting color image.

Author: UPstartDeveloper

.gitignore

@@ -13,6 +13,11 @@ __pycache__/
 # Keras Models
 *model/*
 
+# Images Downloaded for Testing
+*.jpg
+*.jpeg
+*.png
+
 # Distribution / packaging
 .Python
 build/

app/__init__.py

@@ -15,7 +15,7 @@
 predictor = ImagePredictor.init_from_config_path(predictor_config_path)
 
 
-@app.post("/classify-image/", tags=["Detect Fire in an Image"])
-def create_upload_file(file: UploadFile = File(...)):
+@app.post("/detect-fire/", tags=["Detect Fire in an Image"])
+def make_prediction(file: UploadFile = File(...)):
     """Predicts the possibility that a RBG image contains fire."""
     return predictor.predict_from_file(file.file)

app/main.py

@@ -0,0 +1,28 @@
+import unittest
+
+from fastapi.testclient import TestClient
+import tensorflow as tf
+
+from app.main import app
+
+client = TestClient(app)
+
+class ClassifierAPITest(unittest.TestCase):
+
+    TEST_COLOR_IMG_URL = "https://storage.googleapis.com/download.tensorflow.org/example_images/592px-Red_sunflower.jpg"
+
+    def test_make_prediction_normal(self):
+        '''Client submits a RGB image and gets a valid response.'''
+        # A: get image
+        image_path = tf.keras.utils.get_file(
+            origin=self.TEST_COLOR_IMG_URL, 
+            cache_dir=".", cache_subdir="."
+        )
+        with open(image_path, 'rb') as f:
+            # B: format a mock request
+            files = {"file": (f.name, f, "multipart/form-data")}
+            # C: make a request, and verify a valid prediction is returned
+            response = client.post(
+                '/detect-fire', files=files
+            )
+            assert response.status_code == 307

app/main_test.py

@@ -4,4 +4,5 @@ fastapi==0.49.0
 opencv-python-headless==4.5.3.56
 PyYAML==5.3
 uvicorn
+pytest==6.2.5
 python-multipart