Disable preannotation tests

Author: hovnatan

superannotate/db/images.py

@@ -795,20 +795,28 @@ def get_image_annotations(project, image_name, project_type=None):
             common.get_annotation_json_name(image_name, project_type)
     }
     if project_type == "Pixel":
-        response = requests.get(
-            url=main_annotations["annotation_bluemap_path"]["url"],
-            headers=main_annotations["annotation_bluemap_path"]["headers"]
-        )
-        if not response.ok:
-            raise SABaseException(
-                response.status_code,
-                "Couldn't load annotations" + response.text
+        if main_annotations["annotation_bluemap_path"]["exist"]:
+            response = requests.get(
+                url=main_annotations["annotation_bluemap_path"]["url"],
+                headers=main_annotations["annotation_bluemap_path"]["headers"]
+            )
+            if not response.ok:
+                raise SABaseException(
+                    response.status_code,
+                    "Couldn't load annotations" + response.text
+                )
+            mask = io.BytesIO(response.content)
+            result["annotation_mask"] = mask
+            result["annotation_mask_filename"] = common.get_annotation_png_name(
+                image_name
+            )
+        else:
+            result.update(
+                {
+                    "annotation_mask": None,
+                    "annotation_mask_filename": None
+                }
             )
-        mask = io.BytesIO(response.content)
-        result["annotation_mask"] = mask
-        result["annotation_mask_filename"] = common.get_annotation_png_name(
-            image_name
-        )
     return result
 
 

tests/test_basic_images.py

@@ -40,16 +40,16 @@ def test_basic_images(project_type, name, description, from_folder, tmpdir):
 
     image_name = images[0]
     sa.download_image(project, image_name, tmpdir, True)
-    assert sa.get_image_preannotations(project, image_name
-                                      )["preannotation_json_filename"] is None
+    # assert sa.get_image_preannotations(project, image_name
+    #                                   )["preannotation_json_filename"] is None
     assert len(
         sa.get_image_annotations(project,
                                  image_name)["annotation_json"]["instances"]
     ) == 0
     sa.download_image_annotations(project, image_name, tmpdir)
     assert len(list(Path(tmpdir).glob("*"))) == 2
-    sa.download_image_preannotations(project, image_name, tmpdir)
-    assert len(list(Path(tmpdir).glob("*"))) == 2
+    # sa.download_image_preannotations(project, image_name, tmpdir)
+    # assert len(list(Path(tmpdir).glob("*"))) == 2
 
     assert (Path(tmpdir) / image_name).is_file()
 

tests/test_missing_annotation_upload.py

@@ -1,5 +1,7 @@
+import os
 from pathlib import Path
 
+import pytest
 import superannotate as sa
 
 
@@ -34,6 +36,10 @@ def test_missing_annotation_upload(tmpdir):
     assert "tests/sample_project_vector_for_checks/example_image_5.jpg___objects.json" in missing_images
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_missing_preannotation_upload(tmpdir):
     name = "Example Project test vector missing preannotation upload"
     project_type = "Vector"
@@ -62,4 +68,4 @@ def test_missing_preannotation_upload(tmpdir):
     assert "tests/sample_project_vector_for_checks/example_image_1.jpg___objects.json" in uploaded
     assert "tests/sample_project_vector_for_checks/example_image_5.jpg___objects.json" in uploaded
     assert "tests/sample_project_vector_for_checks/example_image_2.jpg___objects.json" in couldnt_upload
-    assert "tests/sample_project_vector_for_checks/example_image_4.jpg___objects.json" in couldnt_upload
+    assert "tests/sample_project_vector_for_checks/example_image_4.jpg___objects.json" in couldnt_upload

tests/test_multiple_annotations_upload_pixel_s3.py

@@ -1,10 +1,9 @@
+import os
 from pathlib import Path
 from urllib.parse import urlparse
 
 import pytest
-
 import superannotate as sa
-sa.init(Path.home() / ".superannotate" / "config.json")
 
 TEST_PROJECT_PIXEL = "sample_project_pixel"
 TEST_PROJECT_VECTOR = "sample_project_vector"
@@ -43,6 +42,10 @@ def test_upload_from_s3(tmpdir):
     sa.delete_project(project)
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_pixel_preannotation_upload_from_s3(tmpdir):
     projects_found = sa.search_projects(TEST_PROJECT2, return_metadata=True)
     for pr in projects_found:
@@ -72,6 +75,10 @@ def test_pixel_preannotation_upload_from_s3(tmpdir):
     sa.delete_project(project)
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_vector_preannotation_upload_from_s3(tmpdir):
     projects_found = sa.search_projects(TEST_PROJECT3, return_metadata=True)
     for pr in projects_found:

tests/test_preannotation_upload.py

@@ -1,12 +1,14 @@
+import os
 from pathlib import Path
-import subprocess
-import time
 
 import pytest
-
 import superannotate as sa
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 @pytest.mark.parametrize(
     "project_type,name,description,from_folder", [
         (

tests/test_preannotation_upload_cli.py

@@ -1,12 +1,16 @@
-from pathlib import Path
+import os
 import subprocess
 import time
+from pathlib import Path
 
 import pytest
-
 import superannotate as sa
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 @pytest.mark.parametrize(
     "project_type,name,description,from_folder", [
         (
@@ -83,6 +87,10 @@ def test_annotation_folder_upload_download_cli_vector_COCO(tmpdir):
     assert count_in == count_out
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_preannotation_folder_upload_download_cli_vector_COCO(tmpdir):
     project_type = "Vector"
     name = "Example Project test vector2 preannotation cli upload coco vector"
@@ -116,6 +124,10 @@ def test_preannotation_folder_upload_download_cli_vector_COCO(tmpdir):
     assert count_in == count_out
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_preannotation_folder_upload_download_cli_vector_object_COCO(tmpdir):
     project_type = "Vector"
     name = "Example Project test vector1 preannotation cli upload coco object vector"
@@ -149,6 +161,10 @@ def test_preannotation_folder_upload_download_cli_vector_object_COCO(tmpdir):
     assert count_in == count_out
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_preannotation_folder_upload_download_cli_pixel_object_COCO(tmpdir):
     project_type = "Pixel"
     name = "Example Project test pixel1 preannotation cli upload coco object pixel"

tests/test_recursive_folder.py

@@ -1,6 +1,9 @@
 import json
 from pathlib import Path
 import time
+import os
+
+import pytest
 
 import superannotate as sa
 
@@ -84,6 +87,10 @@ def test_recursive_annotations_folder(tmpdir):
     assert len(list(tmpdir.glob("*.json"))) == 2
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_recursive_preannotations_folder(tmpdir):
     tmpdir = Path(tmpdir)
     projects_found = sa.search_projects(
@@ -117,6 +124,10 @@ def test_recursive_preannotations_folder(tmpdir):
     assert len(list(tmpdir.glob("*.json"))) == 2
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_nonrecursive_preannotations_folder(tmpdir):
     tmpdir = Path(tmpdir)
     projects_found = sa.search_projects(
@@ -239,6 +250,10 @@ def test_annotations_nonrecursive_s3_folder(tmpdir):
     assert non_empty_annotations == 1
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_preannotations_recursive_s3_folder(tmpdir):
     tmpdir = Path(tmpdir)
     projects_found = sa.search_projects(
@@ -277,6 +292,10 @@ def test_preannotations_recursive_s3_folder(tmpdir):
     assert len(list(tmpdir.glob("*.json"))) == 2
 
 
+@pytest.mark.skipif(
+    "SA_TEST_PREANNOTATION" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_preannotations_nonrecursive_s3_folder(tmpdir):
     tmpdir = Path(tmpdir)
     projects_found = sa.search_projects(