add test case involving MAL for pdf

Author: jtsodapop

tests/data/assets/loremipsum.pdf renamed to tests/assets/loremipsum.pdf

@@ -121,6 +121,23 @@ def configured_project(client, ontology, rand_gen, image_url):
     dataset.delete()
 
 
+@pytest.fixture
+def configured_project_pdf(client, ontology, rand_gen, pdf_url):
+    project = client.create_project(name=rand_gen(str))
+    dataset = client.create_dataset(name=rand_gen(str))
+    editor = list(
+        client.get_labeling_frontends(
+            where=LabelingFrontend.name == "editor"))[0]
+    project.setup(editor, ontology)
+    data_row_ids = []
+    data_row_ids.append(dataset.create_data_row(row_data=pdf_url).uid)
+    project.datasets.connect(dataset)
+    project.data_row_ids = data_row_ids
+    yield project
+    project.delete()
+    dataset.delete()
+
+
 @pytest.fixture
 def prediction_id_mapping(configured_project):
     #Maps tool types to feature schema ids

tests/integration/annotation_import/conftest.py

@@ -1,10 +1,12 @@
 import uuid
 import ndjson
 import pytest
+import random
 
 from labelbox.exceptions import MALValidationError, UuidError
 from labelbox.schema.bulk_import_request import BulkImportRequest
 from labelbox.schema.enums import BulkImportRequestState
+from labelbox.schema.annotation_import import MALPredictionImport
 """
 - Here we only want to check that the uploads are calling the validation
 - Then with unit tests we can check the types of errors raised
@@ -185,3 +187,37 @@ def test_delete(configured_project, predictions):
     bulk_import_request.delete()
     all_import_requests = configured_project.bulk_import_requests()
     assert len(list(all_import_requests)) == 0
+
+
+def test_pdf_mal_bbox(client, configured_project_pdf):
+    """
+    tests pdf mal against only a bbox annotation
+    """
+    annotations = []
+    num_annotations = 1
+
+    for row in configured_project_pdf.export_queued_data_rows():
+        for _ in range(num_annotations):
+            annotations.append({
+                "uuid": str(uuid.uuid4()),
+                "name": "bbox",
+                "dataRow": {
+                    "id": row['id']
+                },
+                "bbox": {
+                    "top": round(random.uniform(0, 300), 2),
+                    "left": round(random.uniform(0, 300), 2),
+                    "height": round(random.uniform(200, 500), 2),
+                    "width": round(random.uniform(0, 200), 2)
+                },
+                "page": random.randint(0, 1),
+                "unit": "POINTS"
+            })
+    import_annotations = MALPredictionImport.create_from_objects(
+        client=client,
+        project_id=configured_project_pdf.uid,
+        name=f"import {str(uuid.uuid4())}",
+        predictions=annotations)
+    import_annotations.wait_until_done()
+
+    assert import_annotations.errors == []

tests/integration/annotation_import/test_bulk_import_request.py

@@ -152,6 +152,14 @@ def image_url(client):
                               sign=True)
 
 
+@pytest.fixture(scope="session")
+def pdf_url(client):
+    return client.upload_file('tests/assets/loremipsum.pdf')
+    #   content_type="application/pdf",
+    #   filename="document.pdf",
+    #   sign=True)
+
+
 @pytest.fixture
 def project(client, rand_gen):
     project = client.create_project(name=rand_gen(str))

tests/integration/conftest.py

@@ -183,7 +183,7 @@ def test_same_ontology_after_instructions(
         configured_project_with_complex_ontology):
     project, _ = configured_project_with_complex_ontology
     initial_ontology = project.ontology().normalized
-    project.upsert_instructions('tests/data/assets/loremipsum.pdf')
+    project.upsert_instructions('tests/assets/loremipsum.pdf')
     updated_ontology = project.ontology().normalized
 
     instructions = updated_ontology.pop('projectInstructions')