fix to test cases

Author: jtsodapop

tests/data/serialization/labelbox_v1/test_text.py

@@ -10,6 +10,7 @@ def test_text():
     serialized = next(LBV1Converter.serialize(collection))
 
     payload['media_type'] = 'text'
+    payload['Global Key'] = None
 
     assert serialized.keys() == payload.keys()
     for key in serialized:
@@ -18,6 +19,8 @@ def test_text():
         elif key == 'Label':
             for annotation_a, annotation_b in zip(serialized[key]['objects'],
                                                   payload[key]['objects']):
+                annotation_b['page'] = None
+                annotation_b['unit'] = None
                 if not len(annotation_a['classifications']):
                     # We don't add a classification key to the payload if there is no classifications.
                     annotation_a.pop('classifications')

tests/data/serialization/labelbox_v1/test_unknown_media.py

@@ -16,6 +16,7 @@ def test_image():
 
     for row in payload:
         row['media_type'] = 'image'
+        row['Global Key'] = None
 
     collection = LBV1Converter.deserialize(payload)
     for idx, serialized in enumerate(LBV1Converter.serialize(collection)):
@@ -30,6 +31,8 @@ def test_image():
                     if not len(annotation_a['classifications']):
                         # We don't add a classification key to the payload if there is no classifications.
                         annotation_a.pop('classifications')
+                    annotation_b['page'] = None
+                    annotation_b['unit'] = None
 
                     if isinstance(annotation_b.get('classifications'),
                                   list) and len(

tests/data/serialization/labelbox_v1/test_video.py

@@ -18,6 +18,7 @@ def test_video():
     collection = LBV1Converter.deserialize([payload])
     serialized = next(LBV1Converter.serialize(collection))
     payload['media_type'] = 'video'
+    payload['Global Key'] = None
     assert serialized.keys() == payload.keys()
     for key in serialized:
         if key != 'Label':
@@ -32,6 +33,8 @@ def test_video():
 
                 for obj_a, obj_b in zip(annotation_a['objects'],
                                         annotation_b['objects']):
+                    obj_b['page'] = None
+                    obj_b['unit'] = None
                     obj_a = round_dict(obj_a)
                     obj_b = round_dict(obj_b)
                     assert obj_a == obj_b