Added and passing async tests

Author: Claire Pajot

tests/test_annotation.py

@@ -402,3 +402,19 @@ def test_box_gt_deletion(dataset):
         job_status = job.status()
         assert job_status["status"] == "Completed"
         assert job_status["job_id"] == job.id
+
+    @pytest.mark.integration
+    def test_category_gt_deletion(dataset):
+        annotation = CategoryAnnotation.from_json(TEST_CATEGORY_ANNOTATIONS[0])
+
+        print(annotation)
+
+        response = dataset.annotate(annotations=[annotation])
+
+        assert response["annotations_processed"] == 1
+
+        job = dataset.delete_annotations()
+        job.sleep_until_complete()
+        job_status = job.status()
+        assert job_status["status"] == "Completed"
+        assert job_status["job_id"] == job.id

tests/test_dataset.py

@@ -365,9 +365,10 @@ def test_annotate_async(dataset: Dataset):
     semseg = SegmentationAnnotation.from_json(TEST_SEGMENTATION_ANNOTATIONS[0])
     polygon = PolygonAnnotation.from_json(TEST_POLYGON_ANNOTATIONS[0])
     bbox = BoxAnnotation(**TEST_BOX_ANNOTATIONS[0])
+    category = CategoryAnnotation.from_json(TEST_CATEGORY_ANNOTATIONS[0])
 
     job: AsyncJob = dataset.annotate(
-        annotations=[semseg, polygon, bbox],
+        annotations=[semseg, polygon, bbox, category],
         asynchronous=True,
     )
     job.sleep_until_complete()
@@ -377,11 +378,11 @@ def test_annotate_async(dataset: Dataset):
         "message": {
             "annotation_upload": {
                 "epoch": 1,
-                "total": 2,
+                "total": 3,
                 "errored": 0,
                 "ignored": 0,
                 "datasetId": dataset.id,
-                "processed": 2,
+                "processed": 3,
             },
             "segmentation_upload": {
                 "ignored": 0,
@@ -390,8 +391,8 @@ def test_annotate_async(dataset: Dataset):
             },
         },
         "job_progress": "1.00",
-        "completed_steps": 3,
-        "total_steps": 3,
+        "completed_steps": 4,
+        "total_steps": 4,
     }
 
 
@@ -400,11 +401,12 @@ def test_annotate_async_with_error(dataset: Dataset):
     dataset.append(make_dataset_items())
     semseg = SegmentationAnnotation.from_json(TEST_SEGMENTATION_ANNOTATIONS[0])
     polygon = PolygonAnnotation.from_json(TEST_POLYGON_ANNOTATIONS[0])
+    category = CategoryAnnotation.from_json(TEST_CATEGORY_ANNOTATIONS[0])
     bbox = BoxAnnotation(**TEST_BOX_ANNOTATIONS[0])
     bbox.reference_id = "fake_garbage"
 
     job: AsyncJob = dataset.annotate(
-        annotations=[semseg, polygon, bbox],
+        annotations=[semseg, polygon, bbox, category],
         asynchronous=True,
     )
     job.sleep_until_complete()
@@ -415,11 +417,11 @@ def test_annotate_async_with_error(dataset: Dataset):
         "message": {
             "annotation_upload": {
                 "epoch": 1,
-                "total": 2,
+                "total": 3,
                 "errored": 1,
                 "ignored": 0,
                 "datasetId": dataset.id,
-                "processed": 1,
+                "processed": 2,
             },
             "segmentation_upload": {
                 "ignored": 0,
@@ -428,8 +430,8 @@ def test_annotate_async_with_error(dataset: Dataset):
             },
         },
         "job_progress": "1.00",
-        "completed_steps": 3,
-        "total_steps": 3,
+        "completed_steps": 4,
+        "total_steps": 4,
     }
 
     assert "Item with id fake_garbage doesn" in str(job.errors())