added pytest

run.sh

@@ -1 +1 @@
-streamlit run src/app.py
+streamlit run src/app.py

setup.py

@@ -4,19 +4,19 @@ with open("README.md", "r") as fh:
     long_description = fh.read()
 
 setuptools.setup(
-    name="albumentations-demo", # Replace with your own username
+    name="albumentations-demo",  # Replace with your own username
     version="0.0.1",
     author="Ilya Larchenko",
     author_email="[email protected]",
-    description="Service for demonstration of ",
+    description="Service for demonstration of Albumentations",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    url="https://github.com/pypa/sampleproject",
+    url="https://github.com/IliaLarchenko/albumentations-demo",
     packages=setuptools.find_packages(),
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
-    python_requires='>=3.6',
-)
+    python_requires=">=3.6",
+)

src/app.py

@@ -39,10 +39,6 @@ param_values = show_transform_control(augmentations[transform_name])
 transform = getattr(A, transform_name)(**param_values)
 augmented_image = transform(image=image)["image"]
 
-# show the params passed
-# st.text("Params passed: ")
-# st.code(get_params_string(param_values))
-# st.text("Press R to update")
 
 # show the images
 width_original = 400

src/utils.py

@@ -5,6 +5,14 @@ import json
 import streamlit as st
 
 
+@st.cache
+def get_images_list(path_to_folder: str) -> list:
+    image_names_list = [
+        x for x in os.listdir(path_to_folder) if x[-3:] in ["jpg", "peg", "png"]
+    ]
+    return image_names_list
+
+
 @st.cache
 def load_image(
     image_name: str, path_to_folder: str = "../images", bgr2rgb: bool = True
@@ -17,11 +25,14 @@ def load_image(
 
 
 @st.cache
-def get_images_list(path_to_folder: str) -> list:
-    image_names_list = [
-        x for x in os.listdir(path_to_folder) if x[-3:] in ["jpg", "peg", "png"]
-    ]
-    return image_names_list
+def load_augmentations_config(
+    placeholder_params: dict, path_to_config: str = "configs/augmentations.json"
+) -> dict:
+    with open(path_to_config, "r") as config_file:
+        augmentations = json.load(config_file)
+    for name, params in augmentations.items():
+        params = [fill_placeholders(param, placeholder_params) for param in params]
+    return augmentations
 
 
 def fill_placeholders(params, placeholder_params):
@@ -45,17 +56,6 @@ def fill_placeholders(params, placeholder_params):
     return params
 
 
-@st.cache
-def load_augmentations_config(
-    placeholder_params: dict, path_to_config: str = "configs/augmentations.json"
-) -> dict:
-    with open(path_to_config, "r") as config_file:
-        augmentations = json.load(config_file)
-    for name, params in augmentations.items():
-        params = [fill_placeholders(param, placeholder_params) for param in params]
-    return augmentations
-
-
 def get_params_string(param_values: dict) -> str:
     params_string = ", ".join(
         [k + "=" + str(param_values[k]) for k in param_values.keys()]

src/visuals.py

@@ -15,7 +15,7 @@ def select_image(path_to_images: str = "images"):
     return image
 
 
-def show_transform_control(transform_params: dict):
+def show_transform_control(transform_params: dict) -> dict:
     param_values = {"p": 1.0}
     if len(transform_params) == 0:
         st.sidebar.text("Transform has no parameters")

tests/test_app.py

@@ -1,9 +0,0 @@
-# test
-
-
-def func(x):
-    return x + 1
-
-
-def test_answer():
-    assert func(3) == 4

tests/test_utils.py

@@ -1,6 +1,9 @@
 # test
 
-from src.utils import get_images_list
+import numpy as np
+import albumentations as A
+
+from src.utils import get_images_list, load_image, load_augmentations_config
 
 
 def test_get_images_list():
@@ -8,3 +11,40 @@ def test_get_images_list():
     assert isinstance(images_list, list)
     assert len(images_list) > 0
     assert isinstance(images_list[0], str)
+
+
+def test_load_image():
+    images_list = get_images_list("images")
+    for image_name in images_list:
+        image = load_image(image_name, path_to_folder="images", bgr2rgb=True)
+        assert len(image.shape) == 3, f"error in {image_name}"
+        assert image.shape[2] == 3, f"error in {image_name}"
+        assert image.max() <= 255, f"error in {image_name}"
+        assert image.min() >= 0, f"error in {image_name}"
+
+
+def test_load_augmentations_config():
+    image = np.random.randint(0, 255, (100, 100, 3)).astype(np.uint8)
+    placeholder_params = {
+        "image_width": image.shape[1],
+        "image_height": image.shape[0],
+        "image_half_width": int(image.shape[1] / 2),
+        "image_half_height": int(image.shape[0] / 2),
+    }
+    augmentations = load_augmentations_config(
+        placeholder_params, path_to_config="configs/augmentations.json"
+    )
+
+    for transform_name in augmentations.keys():
+        if transform_name in ["CenterCrop", "RandomCrop"]:
+            param_values = {"p": 1.0, "height": 10, "width": 10}
+        elif transform_name in ["Crop"]:
+            param_values = {"p": 1.0, "x_max": 10, "y_max": 10}
+        else:
+            param_values = {"p": 1.0}
+        transform = getattr(A, transform_name)(**param_values)
+        transformed_image = transform(image=image)["image"]
+        assert len(transformed_image.shape) == 3, f"error in {str(transform)}"
+        assert transformed_image.shape[2] == 3, f"error in {str(transform)}"
+        assert transformed_image.max() <= 255, f"error in {str(transform)}"
+        assert transformed_image.min() >= 0, f"error in {str(transform)}"