update test get envvar

streamlit_img_label/manage.py

@@ -14,8 +14,6 @@
 from PIL import Image
 
 
-
-
 class ImageManager:
     """ImageManager Manage the image object.
 
@@ -28,17 +26,9 @@ def __init__(self, filename):
         self._filename = filename
         self._img = Image.open(filename)
         self._rects = []
-        # self._load_rects()
         self._resized_ratio_w = 1
         self._resized_ratio_h = 1
 
-
-    # def _load_rects(self):
-    #     rects_xml = read_xml(self._filename)
-    #     if rects_xml:
-    #         self._rects = rects_xml
-
-
     def get_img(self):
         """get the image object
 
@@ -47,7 +37,6 @@ def get_img(self):
         """
         return self._img
 
-
     def get_rects(self):
         """get the rects
 
@@ -56,7 +45,6 @@ def get_rects(self):
         """
         return self._rects
 
-
     def resizing_img(self, max_height=700, max_width=700):
         """resizing the image by max_height and max_width.
 
@@ -82,7 +70,6 @@ def resizing_img(self, max_height=700, max_width=700):
         self._resized_ratio_h = self._img.height / resized_img.height
         return resized_img
 
-
     def _resize_rect(self, rect):
         resized_rect = {}
         resized_rect["left"] = rect["left"] / self._resized_ratio_w
@@ -93,7 +80,6 @@ def _resize_rect(self, rect):
             resized_rect["label"] = rect["label"]
         return resized_rect
 
-
     def get_resized_rects(self):
         """get resized the rects according to the resized image.
 
@@ -102,7 +88,6 @@ def get_resized_rects(self):
         """
         return [self._resize_rect(rect) for rect in self._rects]
 
-
     def _chop_box_img(self, rect):
         rect["left"] = int(rect["left"] * self._resized_ratio_w)
         rect["width"] = int(rect["width"] * self._resized_ratio_w)
@@ -126,7 +111,6 @@ def _chop_box_img(self, rect):
             label = rect["label"]
         return (Image.fromarray(prev_img), label)
 
-
     def init_annotation(self, rects):
         """init annotation for current rects.
 
@@ -138,7 +122,6 @@ def init_annotation(self, rects):
         self._current_rects = rects
         return [self._chop_box_img(rect) for rect in self._current_rects]
 
-
     def set_annotation(self, index, label):
         """set the label of the image.
 
@@ -149,8 +132,6 @@ def set_annotation(self, index, label):
         self._current_rects[index]["label"] = label
 
 
-
-
 class ArchiveImageManager:
     """ArchiveImageManager Manage the image object in bytes.
 
@@ -160,13 +141,11 @@ class ArchiveImageManager:
 
     def __init__(self, img):
         """initiate module"""
-        # self._filename = filename
         self._img = Image.open(BytesIO(img))
         self._rects = []
         self._resized_ratio_w = 1
         self._resized_ratio_h = 1
 
-
     def get_img(self):
         """get the image object
 
@@ -175,7 +154,6 @@ def get_img(self):
         """
         return self._img
 
-
     def get_rects(self):
         """get the rects
 
@@ -184,7 +162,6 @@ def get_rects(self):
         """
         return self._rects
 
-
     def resizing_img(self, max_height=400, max_width=400):
         """resizing the image by max_height and max_width.
 
@@ -210,7 +187,6 @@ def resizing_img(self, max_height=400, max_width=400):
         self._resized_ratio_h = self._img.height / resized_img.height
         return resized_img
 
-
     def _resize_rect(self, rect):
         resized_rect = {}
         resized_rect["left"] = rect["left"] / self._resized_ratio_w
@@ -221,7 +197,6 @@ def _resize_rect(self, rect):
             resized_rect["label"] = rect["label"]
         return resized_rect
 
-
     def get_resized_rects(self):
         """get resized the rects according to the resized image.
 
@@ -230,7 +205,6 @@ def get_resized_rects(self):
         """
         return [self._resize_rect(rect) for rect in self._rects]
 
-
     def _chop_box_img(self, rect):
         rect["left"] = int(rect["left"] * self._resized_ratio_w)
         rect["width"] = int(rect["width"] * self._resized_ratio_w)
@@ -253,25 +227,3 @@ def _chop_box_img(self, rect):
         if "label" in rect:
             label = rect["label"]
         return (Image.fromarray(prev_img), label)
-
-
-    def init_annotation(self, rects):
-        """init annotation for current rects.
-
-        Args:
-            rects(list): the bounding boxes of the image.
-        Returns:
-            prev_img(list): list of preview images with default label.
-        """
-        self._current_rects = rects
-        return [self._chop_box_img(rect) for rect in self._current_rects]
-
-
-    def set_annotation(self, index, label):
-        """set the label of the image.
-
-        Args:
-            index(int): the index of the list of bounding boxes of the image.
-            label(str): the label of the bounding box
-        """
-        self._current_rects[index]["label"] = label

tests/test_model.py

@@ -7,29 +7,6 @@
 
 import os
 
-from dotenv import load_dotenv
-
-
-load_dotenv()
-
-
-class TestEnvVar:
-
-    def test_get_envvar(self):
-        api_url = os.getenv('API_URL')
-        dagshub_repo_owner = os.getenv('DAGSHUB_REPO_OWNER')
-        dagshub_repo = os.getenv('DAGSHUB_REPO')
-        model_uri = os.getenv('MODEL_URI')
-
-        print(f"API URL: {api_url}")
-        print(f"Dagshub Repo Owner: {dagshub_repo_owner}")
-        print(f"Dagshub Repo: {dagshub_repo}")
-        print(f"Model URI: {model_uri}")
-
-        assert api_url is not None
-        assert dagshub_repo_owner is not None
-        assert dagshub_repo is not None
-        assert model_uri is not None
 
 
 

tests/test_required_elements.py

@@ -7,6 +7,12 @@
 
 import os
 
+from dotenv import load_dotenv
+
+
+load_dotenv()
+
+
 class TestRequiredElements:
 
     def test_folder_exists(self):
@@ -15,7 +21,6 @@ def test_folder_exists(self):
         for folder in required_folders:
             assert os.path.exists(folder)
 
-
     def test_file_exists(self):
         required_files = ['asset/icn_save.png', 'asset/icn_shield.png',
                           'asset/logo_landing.png', 'asset/logo_aissyr_S.png', 
@@ -32,3 +37,12 @@ def test_file_exists(self):
 
         for file in required_files:
             assert os.path.exists(file)
+
+
+class TestEnvVar:
+
+    def test_get_envvar(self):
+        assert os.getenv('API_URL') is not None
+        assert os.getenv('DAGSHUB_REPO_OWNER') is not None
+        assert os.getenv('DAGSHUB_REPO') is not None
+        assert os.getenv('MODEL_URI') is not None