Tests update

Author: VaghinakDev

src/superannotate/lib/core/serviceproviders.py

@@ -331,15 +331,6 @@ async def list_small_annotations(
     ) -> List[dict]:
         raise NotImplementedError
 
-    @abstractmethod
-    def sort_items_by_size(
-        self,
-        project: entities.ProjectEntity,
-        folder: entities.FolderEntity,
-        item_ids: List[int],
-    ) -> Dict[str, List]:
-        raise NotImplementedError
-
     @abstractmethod
     def get_upload_chunks(
         self,

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

@@ -1737,7 +1737,6 @@ def execute(self):
                         )
                     )
                 except Exception as e:
-                    raise e
                     logger.error(e)
                     self._response.errors = AppException("Can't get annotations.")
                     return self._response

src/superannotate/lib/infrastructure/services/annotation.py

@@ -153,35 +153,6 @@ async def list_small_annotations(
             params=query_params,
         )
 
-    def sort_items_by_size(
-        self,
-        project: entities.ProjectEntity,
-        folder: entities.FolderEntity,
-        item_ids: List[int],
-    ) -> Dict[str, List]:
-        chunk_size = 2000
-        query_params = {
-            "project_id": project.id,
-            "folder_id": folder.id,
-        }
-
-        response_data = {"small": [], "large": []}
-        for i in range(0, len(item_ids), chunk_size):
-            body = {
-                "item_ids": item_ids[i : i + chunk_size],  # noqa
-            }  # noqa
-            response = self.client.request(
-                url=urljoin(self.assets_provider_url, self.URL_CLASSIFY_ITEM_SIZE),
-                method="POST",
-                params=query_params,
-                data=body,
-            )
-            if not response.ok:
-                raise AppException(response.error)
-            response_data["small"].extend(response.data.get("small", []))
-            response_data["large"].extend(response.data.get("large", []))
-        return response_data
-
     def get_upload_chunks(
         self,
         project: entities.ProjectEntity,

tests/unit/test_async_functions.py

@@ -3,7 +3,6 @@
 
 from superannotate import SAClient
 
-
 sa = SAClient()
 
 
@@ -58,24 +57,26 @@ async def _test():
         asyncio.run(_test())
 
     def test_create_task_get_annotations_in_running_event_loop(self):
-      async def _test():
-          task1 = asyncio.create_task(self.nested())
-          task2 = asyncio.create_task(self.nested())
-          await task1
-          await task2
+        async def _test():
+            task1 = asyncio.create_task(self.nested())
+            task2 = asyncio.create_task(self.nested())
+            await task1
+            await task2
 
         asyncio.run(_test())
 
     def test_gather_get_annotations_in_running_event_loop(self):
         async def gather_test():
             await asyncio.gather(self.nested(), self.nested())
+
         asyncio.run(gather_test())
 
     def test_gather_async_for(self):
         async def gather_test():
             async for _ in DummyIterator(delay=0.01, to=2):
                 annotations = sa.get_annotations(TestAsyncFunctions.PROJECT_NAME)
                 assert len(annotations) == 4
+
         asyncio.run(gather_test())
 
     def test_upload_annotations_in_running_event_loop(self):