add is_anomalous attribute to labels in test cases

openvinotoolkit · May 30, 2022 · 16f2138 · 16f2138
1 parent 2aab8df
commit 16f2138
Showing 1 changed file with 32 additions and 6 deletions.
diff --git a/ote_sdk/ote_sdk/tests/usecases/exportable_code/test_prediction_to_annotation_converter.py b/ote_sdk/ote_sdk/tests/usecases/exportable_code/test_prediction_to_annotation_converter.py
@@ -293,7 +293,10 @@ def test_create_converter(self):
                 name="Normal", domain=Domain.ANOMALY_CLASSIFICATION, id=ID("1")
             ),
             LabelEntity(
-                name="Anomalous", domain=Domain.ANOMALY_CLASSIFICATION, id=ID("2")
+                name="Anomalous",
+                domain=Domain.ANOMALY_CLASSIFICATION,
+                id=ID("2"),
+                is_anomalous=True,
             ),
         ]
         label_group = LabelGroup(
@@ -310,7 +313,12 @@ def test_create_converter(self):
         # "ANOMALY_DETECTION" is specified as "converter_type"
         labels = [
             LabelEntity(name="Normal", domain=Domain.ANOMALY_DETECTION, id=ID("1")),
-            LabelEntity(name="Anomalous", domain=Domain.ANOMALY_DETECTION, id=ID("2")),
+            LabelEntity(
+                name="Anomalous",
+                domain=Domain.ANOMALY_DETECTION,
+                id=ID("2"),
+                is_anomalous=True,
+            ),
         ]
         label_group = LabelGroup(name="Anomaly detection labels group", labels=labels)
         label_schema = LabelSchemaEntity(label_groups=[label_group])
@@ -325,7 +333,10 @@ def test_create_converter(self):
         labels = [
             LabelEntity(name="Normal", domain=Domain.ANOMALY_SEGMENTATION, id=ID("1")),
             LabelEntity(
-                name="Anomalous", domain=Domain.ANOMALY_SEGMENTATION, id=ID("2")
+                name="Anomalous",
+                domain=Domain.ANOMALY_SEGMENTATION,
+                id=ID("2"),
+                is_anomalous=True,
             ),
         ]
         label_group = LabelGroup(name="Anomaly detection labels group", labels=labels)
@@ -947,8 +958,18 @@ def test_anomaly_classification_to_annotation_init(
             non_empty_labels = [
                 LabelEntity(name="Normal", domain=Domain.CLASSIFICATION, id=ID("1")),
                 LabelEntity(name="Normal", domain=Domain.CLASSIFICATION, id=ID("2")),
-                LabelEntity(name="Anomalous", domain=Domain.CLASSIFICATION, id=ID("1")),
-                LabelEntity(name="Anomalous", domain=Domain.CLASSIFICATION, id=ID("2")),
+                LabelEntity(
+                    name="Anomalous",
+                    domain=Domain.CLASSIFICATION,
+                    id=ID("1"),
+                    is_anomalous=True,
+                ),
+                LabelEntity(
+                    name="Anomalous",
+                    domain=Domain.CLASSIFICATION,
+                    id=ID("2"),
+                    is_anomalous=True,
+                ),
             ]
             label_group = LabelGroup(
                 name="Classification labels group", labels=non_empty_labels
@@ -1030,7 +1051,12 @@ def check_annotation(actual_annotation: Annotation, expected_labels: list):
 
         non_empty_labels = [
             LabelEntity(name="Normal", domain=Domain.CLASSIFICATION, id=ID("1")),
-            LabelEntity(name="Anomalous", domain=Domain.CLASSIFICATION, id=ID("2")),
+            LabelEntity(
+                name="Anomalous",
+                domain=Domain.CLASSIFICATION,
+                id=ID("2"),
+                is_anomalous=True,
+            ),
         ]
         label_group = LabelGroup(
             name="Anomaly classification labels group", labels=non_empty_labels