update to remove some unnecessary parts of commit

Author: jtsodapop

tests/data/serialization/ndjson/test_export_video_objects.py

@@ -585,10 +585,6 @@ def video_serialized_bbox_label():
     }
 
 
-#ignore uuid because we randomize if there was none
-IGNORE_KEYS = ["uuid"]
-
-
 def test_serialize_video_objects():
     label = video_bbox_label()
     serialized_labels = NDJsonConverter.serialize([label])
@@ -597,7 +593,8 @@ def test_serialize_video_objects():
     manual_label = video_serialized_bbox_label()
 
     for key in label.keys():
-        if key not in IGNORE_KEYS:
+        #ignore uuid because we randomize if there was none
+        if key != "uuid":
             assert label[key] == manual_label[key]
 
     assert len(label['segments']) == 2

tests/data/serialization/ndjson/test_image.py

@@ -5,8 +5,6 @@
 from labelbox.data.serialization.ndjson.converter import NDJsonConverter
 from labelbox.data.annotation_types import Mask, Label, ObjectAnnotation, ImageData, MaskData
 
-IGNORE_KEYS = ['classifications']
-
 
 def round_dict(data):
     if isinstance(data, dict):
@@ -28,8 +26,7 @@ def test_image():
     res = NDJsonConverter.deserialize(data).as_list()
     res = list(NDJsonConverter.serialize(res))
     for r in res:
-        for key in IGNORE_KEYS:
-            r.pop(key, None)
+        r.pop('classifications', None)
     assert [round_dict(x) for x in res] == [round_dict(x) for x in data]
 
 
@@ -41,8 +38,7 @@ def test_image_with_name_only():
     res = NDJsonConverter.deserialize(data).as_list()
     res = list(NDJsonConverter.serialize(res))
     for r in res:
-        for key in IGNORE_KEYS:
-            r.pop(key, None)
+        r.pop('classifications', None)
     assert [round_dict(x) for x in res] == [round_dict(x) for x in data]
 
 
@@ -72,9 +68,7 @@ def test_mask():
     res = NDJsonConverter.deserialize(data).as_list()
     res = list(NDJsonConverter.serialize(res))
     for r in res:
-        for key in IGNORE_KEYS:
-            r.pop(key, None)
-
+        r.pop('classifications', None)
     assert [round_dict(x) for x in res] == [round_dict(x) for x in data]
 
 

tests/integration/annotation_import/test_bulk_import_request.py

@@ -213,6 +213,36 @@ def test_pdf_mal_bbox(client, configured_project_pdf):
                 "page": random.randint(0, 1),
                 "unit": "POINTS"
             })
+        annotations.extend([{
+            'name': 'text',
+            'answer': 'the answer to the text question',
+            'uuid': 'fc1913c6-b735-4dea-bd25-c18152a4715f',
+            "dataRow": {
+                "id": row['id']
+            }
+        }, {
+            'name':
+                'checklist',
+            'uuid':
+                '9d7b2e57-d68f-4388-867a-af2a9b233719',
+            "dataRow": {
+                "id": row['id']
+            },
+            'answer': [{
+                'name': 'first_checklist_answer'
+            }, {
+                'name': 'second_checklist_answer'
+            }]
+        }, {
+            'name': 'radio',
+            'answer': {
+                'name': 'second_radio_answer'
+            },
+            'uuid': 'ad60897f-ea1a-47de-b923-459339764921',
+            "dataRow": {
+                "id": row['id']
+            }
+        }])
     import_annotations = MALPredictionImport.create_from_objects(
         client=client,
         project_id=configured_project_pdf.uid,