fix export

Author: VaghinakDev

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

@@ -112,7 +112,7 @@ def execute(self):
                 integration_id=self._integration_id,
             )
             if self._annotation_statuses:
-                kwargs["annotation_statuses"] = (self._annotation_statuses,)
+                kwargs["annotation_statuses"] = self._annotation_statuses
             if self._export_type:
                 kwargs["export_type"] = self._export_type
             response = self._service_provider.prepare_export(**kwargs)

tests/applicatoin/base.py

@@ -0,0 +1,46 @@
+from unittest import TestCase
+
+from src.superannotate import SAClient
+
+
+sa = SAClient()
+
+
+class BaseTestCase(TestCase):
+    PROJECT_NAME = ""
+    PROJECT_DESCRIPTION = "Desc"
+    PROJECT_TYPE = "Vector"
+    TEST_FOLDER_PATH = "data_set"
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        BaseTestCase.PROJECT_NAME = BaseTestCase.__class__.__name__
+
+    def setUp(self, *args, **kwargs):
+        self.tearDown()
+        self._project = sa.create_project(
+            self.PROJECT_NAME, self.PROJECT_DESCRIPTION, self.PROJECT_TYPE
+        )
+
+    def tearDown(self) -> None:
+        try:
+            projects = sa.search_projects(self.PROJECT_NAME, return_metadata=True)
+            for project in projects:
+                try:
+                    sa.delete_project(project)
+                except Exception as e:
+                    print(str(e))
+        except Exception as e:
+            print(str(e))
+
+    def _attach_items(self, count=5, folder=None):
+        path = self.PROJECT_NAME
+        if folder:
+            path = f"{self.PROJECT_NAME}/{folder}"
+        sa.attach_items(
+            path,
+            [
+                {"name": f"example_image_{i}.jpg", "url": f"url_{i}"}
+                for i in range(1, count + 1)
+            ],  # noqa
+        )

tests/integration/export/test_export.py

@@ -111,3 +111,10 @@ def test_upload_s3(self):
         ).get("Contents", []):
             files.append(object_data["Key"])
         self.assertEqual(25, len(files))
+
+    def test_export_with_statuses(self):
+        with tempfile.TemporaryDirectory() as tmpdir_name:
+            export = sa.prepare_export(self.PROJECT_NAME, annotation_statuses=['NotStarted'], include_fuse=True)
+            sa.download_export(self.PROJECT_NAME, export, tmpdir_name)
+            assert not filecmp.dircmp(tmpdir_name, self.TEST_FOLDER_PATH).left_only
+            assert not filecmp.dircmp(tmpdir_name, self.TEST_FOLDER_PATH).right_only