Fix tests

Author: VaghinakDev

src/superannotate/lib/core/usecases/annotations.py

@@ -580,26 +580,25 @@ def get_existing_name_item_mapping(
 
     @property
     def annotation_upload_data(self) -> UploadAnnotationAuthData:
-        CHUNK_SIZE=UploadAnnotationsFromFolderUseCase.CHUNK_SIZE_PATHS
+        CHUNK_SIZE = UploadAnnotationsFromFolderUseCase.CHUNK_SIZE_PATHS
 
         if self._annotation_upload_data:
             return self._annotation_upload_data
 
-        images={}
+        images = {}
         for i in range(0, len(self._item_ids), CHUNK_SIZE):
             tmp = self._service_provider.get_annotation_upload_data(
                 project=self._project,
                 folder=self._folder,
-                item_ids=self._item_ids[i:i+CHUNK_SIZE],
-
+                item_ids=self._item_ids[i : i + CHUNK_SIZE],
             )
             if not tmp.ok:
                 raise AppException(tmp.errors)
             else:
                 images.update(tmp.data.images)
 
-        self._annotation_upload_data=tmp.data
-        self._annotation_upload_data.images=images
+        self._annotation_upload_data = tmp.data
+        self._annotation_upload_data.images = images
 
         return self._annotation_upload_data
 

src/superannotate/lib/core/usecases/images.py

@@ -1046,7 +1046,7 @@ def filter_paths(self, paths: List[str]):
                 raise AppException(response.error)
             image_list.extend([image.name for image in response.data])
 
-        image_list=set(image_list)
+        image_list = set(image_list)
         images_to_upload = []
 
         for path in filtered_paths:

tests/integration/annotations/test_annotations_upload_status_change.py

@@ -38,18 +38,6 @@ def test_upload_annotations_from_folder_to_project__upload_status(self, reporter
             sa.get_item_metadata(self.PROJECT_NAME, self.IMAGE_NAME)["annotation_status"]
         )
 
-    @pytest.mark.flaky(reruns=2)
-    @patch("lib.infrastructure.controller.Reporter")
-    def test_upload_preannotations_from_folder_to_project__upload_status(self, reporter):
-        reporter_mock = MagicMock()
-        reporter.return_value = reporter_mock
-        sa.upload_image_to_project(self.PROJECT_NAME, join(self.folder_path, self.IMAGE_NAME))
-        sa.upload_preannotations_from_folder_to_project(self.PROJECT_NAME, self.folder_path)
-        self.assertEqual(
-            constances.AnnotationStatus.IN_PROGRESS.name,
-            sa.get_item_metadata(self.PROJECT_NAME, self.IMAGE_NAME)["annotation_status"]
-        )
-
     @pytest.mark.flaky(reruns=2)
     @patch("lib.infrastructure.controller.Reporter")
     def test_upload_image_annotations__upload_status(self, reporter):

tests/integration/annotations/test_missing_annotation_upload.py

@@ -45,23 +45,3 @@ def test_missing_annotation_upload(self):
                 ]
             )
         )
-
-    def test_missing_pre_annotation_upload(self):
-
-        sa.upload_images_from_folder_to_project(
-            self.PROJECT_NAME, self.folder_path, annotation_status="NotStarted"
-        )
-        sa.create_annotation_classes_from_classes_json(
-            self.PROJECT_NAME, f"{self.folder_path}/classes/classes.json"
-        )
-        (
-            uploaded,
-            could_not_upload,
-            missing_images,
-        ) = sa.upload_preannotations_from_folder_to_project(
-            self.PROJECT_NAME, self.folder_path
-        )
-        self.assertEqual(len(uploaded), 3)
-        self.assertEqual(len(could_not_upload), 0)
-        self.assertEqual(len(missing_images), 1)
-

tests/integration/test_basic_images.py

@@ -94,18 +94,6 @@ def folder_path(self):
     def classes_json_path(self):
         return f"{self.folder_path}/classes/classes.json"
 
-    def test_vector_annotations_with_tag_folder_upload_preannotation(self):
-        sa.upload_images_from_folder_to_project(
-            self.PROJECT_NAME, self.folder_path, annotation_status="InProgress"
-        )
-        sa.create_annotation_classes_from_classes_json(
-            self.PROJECT_NAME, self.classes_json_path
-        )
-        uploaded_annotations, _, _ = sa.upload_preannotations_from_folder_to_project(
-            self.PROJECT_NAME, self.folder_path
-        )
-        self.assertEqual(len(uploaded_annotations), 1)
-
 
 class TestPixelImages(BaseTestCase):
     PROJECT_NAME = "sample_project_pixel"

tests/integration/test_cli.py

@@ -127,19 +127,6 @@ def test_upload_export(self):
             self.assertEqual(len(list(test_dir.glob("*.jpg"))), 0)
             self.assertEqual(len(list(test_dir.glob("*.png"))), 0)
 
-    def test_vector_pre_annotation_folder_upload_download_cli(self):
-        self._create_project()
-
-        sa.create_annotation_classes_from_classes_json(
-            self.PROJECT_NAME, f"{self.vector_folder_path}/classes/classes.json"
-        )
-        self.safe_run(self._cli.upload_images, project=self.PROJECT_NAME, folder=str(self.convertor_data_path),
-                      extensions="jpg",
-                      set_annotation_status="QualityCheck")
-        self.safe_run(self._cli.upload_preannotations, project=self.PROJECT_NAME, folder=str(self.convertor_data_path),
-                      format="COCO",
-                      dataset_name="instances_test")
-
     def test_vector_annotation_folder_upload_download_cli(self):
         self._create_project()
         sa.create_annotation_classes_from_classes_json(

tests/integration/test_depricated_functions_document.py

@@ -83,10 +83,6 @@ def test_deprecated_functions(self):
             sa.download_image_annotations(self.PROJECT_NAME, self.UPLOAD_IMAGE_NAME, "./")
         except AppException as e:
             self.assertIn(self.EXCEPTION_MESSAGE, str(e))
-        try:
-            sa.copy_image(self.PROJECT_NAME, self.UPLOAD_IMAGE_NAME, self.PROJECT_NAME_2)
-        except AppException as e:
-            self.assertIn(self.EXCEPTION_MESSAGE, str(e))
         try:
             sa.upload_images_to_project(self.PROJECT_NAME, [self.image_path, ])
         except AppException as e:
@@ -103,10 +99,6 @@ def test_deprecated_functions(self):
             sa.set_project_workflow(self.PROJECT_NAME, [{}])
         except AppException as e:
             self.assertIn(self.EXCEPTION_MESSAGE, str(e))
-        try:
-            sa.upload_preannotations_from_folder_to_project(self.PROJECT_NAME, self.folder_path)
-        except AppException as e:
-            self.assertIn(self.EXCEPTION_MESSAGE, str(e))
         try:
             sa.add_annotation_point_to_image(self.PROJECT_NAME, self.UPLOAD_IMAGE_NAME, [1, 2], "some class")
         except AppException as e:

tests/integration/test_depricated_functions_video.py

@@ -76,10 +76,6 @@ def test_deprecated_functions(self):
             sa.download_image_annotations(self.PROJECT_NAME, self.UPLOAD_IMAGE_NAME, "./")
         except AppException as e:
             self.assertIn(self.EXCEPTION_MESSAGE, str(e))
-        try:
-            sa.copy_image(self.PROJECT_NAME, self.UPLOAD_IMAGE_NAME, self.PROJECT_NAME_2)
-        except AppException as e:
-            self.assertIn(self.EXCEPTION_MESSAGE, str(e))
         try:
             sa.upload_images_to_project(self.PROJECT_NAME, [self.image_path, ])
         except AppException as e:
@@ -96,10 +92,6 @@ def test_deprecated_functions(self):
             sa.set_project_workflow(self.PROJECT_NAME, [{}])
         except AppException as e:
             self.assertIn(self.EXCEPTION_MESSAGE, str(e))
-        try:
-            sa.upload_preannotations_from_folder_to_project(self.PROJECT_NAME, self.folder_path)
-        except AppException as e:
-            self.assertIn(self.EXCEPTION_MESSAGE, str(e))
         try:
             sa.get_project_workflow(self.PROJECT_NAME)
         except AppException as e:

tests/integration/test_limitations.py

@@ -1,21 +1,19 @@
 import os
-from unittest.mock import patch
 from os.path import dirname
+from unittest.mock import patch
 
-from src.superannotate import SAClient
-sa = SAClient()
 from src.superannotate import AppException
+from src.superannotate import SAClient
 from src.superannotate.lib.core import UPLOAD_FOLDER_LIMIT_ERROR_MESSAGE
 from src.superannotate.lib.core import UPLOAD_PROJECT_LIMIT_ERROR_MESSAGE
 from src.superannotate.lib.core import UPLOAD_USER_LIMIT_ERROR_MESSAGE
-from src.superannotate.lib.core import COPY_FOLDER_LIMIT_ERROR_MESSAGE
-from src.superannotate.lib.core import COPY_PROJECT_LIMIT_ERROR_MESSAGE
-from src.superannotate.lib.core import COPY_SUPER_LIMIT_ERROR_MESSAGE
 from tests.integration.base import BaseTestCase
 from tests.moks.limitatoins import folder_limit_response
 from tests.moks.limitatoins import project_limit_response
 from tests.moks.limitatoins import user_limit_response
 
+sa = SAClient()
+
 
 class TestLimitsUploadImagesFromFolderToProject(BaseTestCase):
     PROJECT_NAME = "TestLimitsUploadImagesFromFolderToProject"
@@ -49,43 +47,3 @@ def test_user_limitations(self, *_):
             _, _, __ = sa.upload_images_from_folder_to_project(
                 project=self._project["name"], folder_path=self.folder_path
             )
-
-
-class TestLimitsCopyImage(BaseTestCase):
-    PROJECT_NAME = "TestLimitsCopyImage"
-    PROJECT_DESCRIPTION = "Desc"
-    PROJECT_TYPE = "Vector"
-    TEST_FOLDER_PTH = "data_set"
-    TEST_FOLDER_PATH = "data_set/sample_project_vector"
-    EXAMPLE_IMAGE_1 = "example_image_1.jpg"
-
-    @property
-    def folder_path(self):
-        return os.path.join(dirname(dirname(__file__)), self.TEST_FOLDER_PATH)
-
-    def test_folder_limitations(self):
-        sa.upload_image_to_project(self._project["name"], os.path.join(self.folder_path, self.EXAMPLE_IMAGE_1))
-        sa.create_folder(self._project["name"], self._project["name"])
-        with patch("lib.infrastructure.serviceprovider.ServiceProvider.get_limitations") as limit_response:
-            limit_response.return_value = folder_limit_response
-            with self.assertRaisesRegexp(AppException, COPY_FOLDER_LIMIT_ERROR_MESSAGE):
-                _, _, __ = sa.copy_image(
-                    self._project["name"], self.folder_path, f"{self.PROJECT_NAME}/{self.PROJECT_NAME}")
-
-    def test_project_limitations(self, ):
-        sa.upload_image_to_project(self._project["name"], os.path.join(self.folder_path, self.EXAMPLE_IMAGE_1))
-        sa.create_folder(self._project["name"], self._project["name"])
-        with patch("lib.infrastructure.serviceprovider.ServiceProvider.get_limitations") as limit_response:
-            limit_response.return_value = project_limit_response
-            with self.assertRaisesRegexp(AppException, COPY_PROJECT_LIMIT_ERROR_MESSAGE):
-                _, _, __ = sa.copy_image(
-                    self._project["name"], self.folder_path, f"{self.PROJECT_NAME}/{self.PROJECT_NAME}")
-
-    def test_user_limitations(self, ):
-        sa.upload_image_to_project(self._project["name"], os.path.join(self.folder_path, self.EXAMPLE_IMAGE_1))
-        sa.create_folder(self._project["name"], self._project["name"])
-        with patch("lib.infrastructure.serviceprovider.ServiceProvider.get_limitations") as limit_response:
-            limit_response.return_value = user_limit_response
-            with self.assertRaisesRegexp(AppException, COPY_SUPER_LIMIT_ERROR_MESSAGE):
-                _, _, __ = sa.copy_image(
-                    self._project["name"], self.folder_path, f"{self.PROJECT_NAME}/{self.PROJECT_NAME}")