add test pixel project blue mask

Author: Narek Mkhitaryan

tests/data_set/sample_explore_export_pixel/classes/classes.json

@@ -0,0 +1 @@
+[{"id": 4991071, "project_id": 533166, "name": "cat3", "color": "#d08e91", "count": 0, "type": "object", "createdAt": "2023-07-20T12:58:35.000Z", "updatedAt": "2023-07-20T12:58:35.000Z", "attribute_groups": []}, {"id": 4991070, "project_id": 533166, "name": "cat2", "color": "#1c09ea", "count": 0, "type": "object", "createdAt": "2023-07-20T12:58:30.000Z", "updatedAt": "2023-07-20T12:58:30.000Z", "attribute_groups": []}, {"id": 4991069, "project_id": 533166, "name": "cat 1", "color": "#e23787", "count": 0, "type": "object", "createdAt": "2023-07-20T12:58:24.000Z", "updatedAt": "2023-07-20T12:58:24.000Z", "attribute_groups": []}]

tests/data_set/sample_explore_export_pixel/file_example.jpg

@@ -0,0 +1 @@
+{"metadata": {"height": 560, "width": 1440, "name": "Hero Pedigree Cats.jpg", "lastAction": {"email": "narekm@superannotate.com", "timestamp": 1689857925400}, "projectId": 533166, "isPredicted": false, "isSegmented": false, "status": "Completed", "pinned": false, "annotatorEmail": null, "qaEmail": null}, "instances": [{"classId": 4991070, "probability": 100, "visible": true, "attributes": [], "parts": [{"color": "#00005a"}], "error": null, "locked": false, "createdAt": "2023-07-20T12:59:49.061Z", "createdBy": {"email": "narekm@superannotate.com", "role": "Admin"}, "creationType": "Manual", "updatedAt": "2023-07-20T13:00:00.234Z", "updatedBy": {"email": "narekm@superannotate.com", "role": "Admin"}, "className": "cat2"}, {"classId": 4991069, "probability": 100, "visible": true, "attributes": [], "parts": [{"color": "#00000f"}, {"color": "#00001e"}, {"color": "#00002d"}, {"color": "#00003c"}], "error": null, "locked": false, "createdAt": "2023-07-20T12:59:02.464Z", "createdBy": {"email": "narekm@superannotate.com", "role": "Admin"}, "creationType": "Manual", "updatedAt": "2023-07-20T12:59:38.786Z", "updatedBy": {"email": "narekm@superannotate.com", "role": "Admin"}, "className": "cat 1"}], "comments": [], "tags": []}

tests/data_set/sample_explore_export_pixel/file_example.jpg.json

@@ -1,4 +1,5 @@
 import os
+import tempfile
 from pathlib import Path
 
 from src.superannotate import SAClient
@@ -122,3 +123,48 @@ def test_annotation_folder_upload_download(self):
             self.PROJECT_NAME, self.folder_path
         )
         assert len(uploaded) == 1
+
+
+class TestExportUploadPixel(BaseTestCase):
+    PROJECT_NAME = "Test-TestExportUploadPixel"
+    PROJECT_DESCRIPTION = "Desc"
+    PROJECT_TYPE = "PIXEL"
+    TEST_FOLDER_PATH = "data_set/sample_explore_export_pixel"
+    ITEM_NAME = "file_example.jpg"
+
+    @property
+    def data_set(self):
+        return Path(__file__).parent.parent.parent
+
+    @property
+    def folder_path(self):
+        return os.path.join(Path(__file__).parent.parent.parent, self.TEST_FOLDER_PATH)
+
+    def test_annotation_folder_upload_download(self):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            sa.attach_items(
+                self.PROJECT_NAME,
+                [{"name": self.ITEM_NAME, "url": "url_"}],
+            )
+            sa.create_annotation_classes_from_classes_json(
+                self.PROJECT_NAME, f"{self.folder_path}/classes/classes.json"
+            )
+            uploaded, _, _ = sa.upload_annotations_from_folder_to_project(
+                self.PROJECT_NAME, self.folder_path
+            )
+            assert len(uploaded) == 1
+            export_name = sa.prepare_export(self.PROJECT_NAME)["name"]
+            sa.download_export(
+                project=self.PROJECT_NAME, export=export_name, folder_path=tmpdir
+            )
+            list_dir = os.listdir(tmpdir)
+            assert f"{self.ITEM_NAME}___save.png" in list_dir
+            assert f"{self.ITEM_NAME}.json" in list_dir
+            assert "classes" in list_dir
+
+            with open(f"{tmpdir}/{self.ITEM_NAME}___save.png", "rb") as f1, open(
+                f"{self.folder_path}/{self.ITEM_NAME}___save.png", "rb"
+            ) as f2:
+                contents1 = f1.read()
+                contents2 = f2.read()
+                assert contents1 == contents2