fix OCR group type tests

Author: Narek Mkhitaryan

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

@@ -1,3 +1,4 @@
+import nest_asyncio
 from lib.core.usecases.annotations import *  # noqa: F403 F401
 from lib.core.usecases.classes import *  # noqa: F403 F401
 from lib.core.usecases.custom_fields import *  # noqa: F403 F401
@@ -8,5 +9,4 @@
 from lib.core.usecases.models import *  # noqa: F403 F401
 from lib.core.usecases.projects import *  # noqa: F403 F401
 
-import nest_asyncio
-nest_asyncio.apply()
+nest_asyncio.apply()

tests/integration/classes/test_create_annotation_class.py

@@ -2,7 +2,6 @@
 import tempfile
 
 import pytest
-from lib.core.entities.classes import AttributeGroup
 from src.superannotate import AppException
 from src.superannotate import SAClient
 from tests import DATA_SET_PATH
@@ -252,18 +251,16 @@ def test_create_annotation_class_via_ocr_group_type(self):
             f"OCR attribute group is not supported for project type {self.PROJECT_TYPE}.",
         ):
             attribute_groups = [
-                AttributeGroup(
-                    **{
-                        "id": 21448,
-                        "class_id": 56820,
-                        "name": "Large",
-                        "group_type": "ocr",
-                        "is_multiselect": 0,
-                        "createdAt": "2020-09-29T10:39:39.000Z",
-                        "updatedAt": "2020-09-29T10:39:39.000Z",
-                        "attributes": [],
-                    }
-                )
+                {
+                    "id": 21448,
+                    "class_id": 56820,
+                    "name": "Large",
+                    "group_type": "ocr",
+                    "is_multiselect": 0,
+                    "createdAt": "2020-09-29T10:39:39.000Z",
+                    "updatedAt": "2020-09-29T10:39:39.000Z",
+                    "attributes": [],
+                }
             ]
             sa.create_annotation_class(
                 self.PROJECT_NAME,

tests/unit/test_async_functions.py

@@ -63,18 +63,21 @@ async def _test():
             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
+            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):