Rename `test.py` to `test_model.py`

test/test.py

@@ -1 +0,0 @@
-import pytest

test/test_model.py

@@ -0,0 +1,50 @@
+from PIL import Image
+from tensorflow import keras
+import sys
+import io
+
+sys.path.append("./../app")
+import model
+
+model_instance: keras.Model = model.create_model()
+
+
+def template(filename: str) -> (int, int, Image.Image):
+    image: Image.Image = Image.open(filename)
+    width: int
+    height: int
+    width, height = image.size
+    image_bytes: io.BytesIO = io.BytesIO()
+    image.save(image_bytes, format=image.format)
+    output_image: Image.Image = model.run_model(image_bytes, model_instance)
+    return width, height, output_image
+
+
+def test_image_jpg():
+    width: int
+    height: int
+    output_image: Image.Image
+    width, height, output_image = template("test_images/test1.jpg")
+
+    assert width == output_image.size[0]
+    assert height == output_image.size[1]
+
+
+def test_image_png():
+    width: int
+    height: int
+    output_image: Image.Image
+    width, height, output_image = template("test_images/test2.png")
+
+    assert width == output_image.size[0]
+    assert height == output_image.size[1]
+
+
+def test_image_jpeg():
+    width: int
+    height: int
+    output_image: Image.Image
+    width, height, output_image = template("test_images/test3.jpeg")
+
+    assert width == output_image.size[0]
+    assert height == output_image.size[1]