Added attributes to unit tests

Author: Tim-Kerr

libs/labelbox/tests/unit/test_unit_ontology.py

@@ -4,6 +4,7 @@
 from lbox.exceptions import InconsistentOntologyException
 
 from labelbox import Classification, OntologyBuilder, Option, Tool
+from labelbox.schema.tool_building.types import FeatureSchemaAttribute
 
 _SAMPLE_ONTOLOGY = {
     "tools": [
@@ -34,7 +35,12 @@
             "name": "bbox",
             "color": "#FF0000",
             "tool": "rectangle",
-            "attributes": None,
+            "attributes": [
+                FeatureSchemaAttribute(
+                    attriubteName="auto-ocr",
+                    attributeValue="true"
+                )
+            ],
             "classifications": [
                 {
                     "schemaNodeId": None,
@@ -71,7 +77,12 @@
                             "options": [],
                         },
                     ],
-                    "attributes": None,
+                    "attributes": [
+                        FeatureSchemaAttribute(
+                            attributeName="requires-connection",
+                            attributeValue="true"
+                        )
+                    ],
                 },
                 {
                     "schemaNodeId": None,