MODEL-1489: Rewrote tests

Author: dubininsergey

libs/labelbox/tests/data/annotation_import/test_data_types.py

@@ -1,12 +1,8 @@
-import datetime
-from labelbox.schema.label import Label
 import pytest
-import uuid
 
 from labelbox.data.annotation_types.data import (
     AudioData,
     ConversationData,
-    DicomData,
     DocumentData,
     HTMLData,
     ImageData,
@@ -15,11 +11,8 @@
 from labelbox.data.serialization import NDJsonConverter
 from labelbox.data.annotation_types.data.video import VideoData
 
-import labelbox as lb
 import labelbox.types as lb_types
 from labelbox.schema.media_type import MediaType
-from labelbox.schema.annotation_import import AnnotationImportState
-from labelbox import Project, Client
 
 # Unit test for label based on data type.
 # TODO: Dicom removed it is unstable when you deserialize and serialize on label import. If we intend to keep this library this needs add generic data types tests work with this data type.
@@ -84,20 +77,3 @@ def test_data_row_type_by_global_key(
 
     assert data_label.data.global_key == label.data.global_key
     assert label.annotations == data_label.annotations
-
-
-@pytest.mark.parametrize("_, data_class, annotations", test_params)
-def test_import_label_annotations_with_is_benchmark_reference_flag(
-        data_class, annotations, _):
-    labels = [
-        lb_types.Label(data=data_class(uid=str(uuid.uuid4()),
-                                       url="http://test.com"),
-                       annotations=annotations,
-                       is_benchmark_reference=True)
-    ]
-    serialized_annotations = get_annotation_comparison_dicts_from_labels(labels)
-
-    assert len(serialized_annotations) == len(annotations)
-    for serialized_annotation in serialized_annotations:
-        assert serialized_annotation["isBenchmarkReferenceLabel"]
-

libs/labelbox/tests/data/serialization/ndjson/test_conversation.py

@@ -101,3 +101,18 @@ def test_conversation_entity_import(filename: str):
     res = list(NDJsonConverter.deserialize(data))
     res = list(NDJsonConverter.serialize(res))
     assert res == data
+
+
+def test_benchmark_reference_label_flag():
+    label = lb_types.Label(data=lb_types.ConversationData(global_key='my_global_key'),
+                   annotations=[
+                       lb_types.ClassificationAnnotation(
+                           name='free_text',
+                           message_id="0",
+                           value=lb_types.Text(answer="sample text"))
+                   ],
+                   is_benchmark_reference=True
+                   )
+
+    res = list(NDJsonConverter.serialize([label]))
+    assert res[0]["isBenchmarkReferenceLabel"]

libs/labelbox/tests/data/serialization/ndjson/test_rectangle.py

@@ -85,3 +85,23 @@ def test_rectangle_mixed_start_end_points():
 
     res = list(NDJsonConverter.deserialize(res))
     assert res == [label]
+
+
+def test_benchmark_reference_label_flag():
+    bbox = lb_types.ObjectAnnotation(
+        name="bbox",
+        value=lb_types.Rectangle(
+            start=lb_types.Point(x=81, y=28),
+            end=lb_types.Point(x=38, y=69),
+        ),
+        extra={"uuid": "c1be3a57-597e-48cb-8d8d-a852665f9e72"}
+    )
+
+    label = lb_types.Label(
+        data={"uid":DATAROW_ID},
+        annotations=[bbox],
+        is_benchmark_reference=True
+    )
+
+    res = list(NDJsonConverter.serialize([label]))
+    assert res[0]["isBenchmarkReferenceLabel"]