Tests for image upload error, raising a UploadImageError

Author: Leandro Rosemberg

tests/annotations/invalid_annotation.json

@@ -0,0 +1,8 @@
+{
+    "it is": [
+        {
+            "a": 0,
+            "invalid annotation": true
+        }
+    ]
+}

tests/annotations/valid_annotation.json

@@ -0,0 +1,44 @@
+{
+    "info": {
+        "year": "2020",
+        "version": "1",
+        "description": "None",
+        "contributor": "Linas",
+        "url": "https://app.roboflow.ai/datasets/hard-hat-sample/1",
+        "date_created": "2000-01-01T00:00:00+00:00"
+    },
+    "licenses": [{"id": 1, "url": "https://creativecommons.org/publicdomain/zero/1.0/", "name": "Public Domain"}],
+    "categories": [
+        {"id": 0, "name": "cat", "supercategory": "animals"}
+    ],
+    "images": [
+        {
+            "id": 0,
+            "license": 1,
+            "file_name": "bla.JPG",
+            "height": 1024,
+            "width": 1792,
+            "date_captured": "2020-07-20T19:39:26+00:00"
+        }
+    ],
+    "annotations": [
+        {
+            "id": 0,
+            "image_id": 0,
+            "category_id": 0,
+            "bbox": [45, 2, 85, 85],
+            "area": 7225,
+            "segmentation": [],
+            "iscrowd": 0
+        },
+        {
+            "id": 1,
+            "image_id": 0,
+            "category_id": 0,
+            "bbox": [324, 29, 72, 81],
+            "area": 5832,
+            "segmentation": [],
+            "iscrowd": 0
+        }
+    ]
+}

tests/test_project.py

@@ -2,7 +2,8 @@
 
 from roboflow import API_URL
 from roboflow.config import DEFAULT_BATCH_NAME
-from tests import PROJECT_NAME, ROBOFLOW_API_KEY, RoboflowTest
+from roboflow.core.exceptions import UploadImageError
+from tests import RoboflowTest, PROJECT_NAME, ROBOFLOW_API_KEY
 
 
 class TestProject(RoboflowTest):
@@ -41,3 +42,23 @@ def test_check_valid_image_with_unaccepted_formats(self):
 
         for image in images_to_test:
             self.assertFalse(self.project.check_valid_image(f"tests/images/{image}"))
+
+    def test_upload_image_that_already_is_annotated_raises_upload_image_error(self):
+        responses.add(
+            responses.POST,
+            f"{API_URL}/dataset/{PROJECT_NAME}/upload?api_key={ROBOFLOW_API_KEY}" f"&batch={DEFAULT_BATCH_NAME}",
+            json={
+                "message": "Image was already annotated.",
+                "type": "InvalidImageException",
+                "hint": "This image was already annotated; to overwrite the annotation, pass overwrite=true...",
+              },
+            status=200,
+        )
+
+        with self.assertRaises(UploadImageError) as error:
+            self.project.upload(
+                "tests/images/rabbit.JPG",
+                annotation_path="tests/annotations/valid_annotation.json",
+            )
+
+        self.assertEquals(str(error.exception), "Error uploading image: Image was already annotated.")