add tests

Author: manibox

labelbox/client.py

@@ -906,3 +906,27 @@ def create_feature_schema(self, normalized):
         # But the features are the same so we just grab the feature schema id
         res['id'] = res['normalized']['featureSchemaId']
         return Entity.FeatureSchema(self, res)
+
+    def create_resource_tag(self, tag=None):
+        """
+        Creates a resource tag.
+            >>> tag = {'text': 'tag-1',  'color': 'ffffff'}
+
+        Args:
+            tag (dict): A resource tag {'text': 'tag-1', 'color': 'fffff'}
+        Returns:
+            The created resource tag.
+        """
+        tag_text_param = "text"
+        tag_color_param = "color"
+
+        query_str = """mutation CreateResourceTagPyApi($text:String!,$color:String!) {
+                createResourceTag(input:{text:$%s,color:$%s}) {id,text,color}}
+        """ % (tag_text_param, tag_color_param)
+
+        params = {
+            tag_text_param: tag.get("text", ""),
+            tag_color_param: tag.get("color", "")
+        }
+        res = self.execute(query_str, params)
+        return Entity.ResourceTag(self, res['createResourceTag'])

labelbox/schema/organization.py

@@ -131,16 +131,7 @@ def get_resource_tags(self) -> List["ResourceTag"]:
         """
         Returns all resource tags for an organization
         """
-        res = self.client.execute(
-            """query {
-                organization {
-                    resourceTag {
-                        id,
-                        text,
-                        color
-                    }
-                }
-            }""")
+        res = self.client.execute("""query GetOrganizationResourceTagsPyApi{organization{resourceTag{id,text,color}}}""")
 
         return res['organization']['resourceTag']
 

labelbox/schema/project.py

@@ -132,9 +132,8 @@ def create_project_resource_tags(self, resource_tag_ids: List[str]) -> List[str]
         project_id_param = "projectId"
         tag_ids_param = "resourceTagIds"
 
-        query_str = """mutation AttatchProjectResourceTags($%s: ID!, $%s: [String!]) {
-            project(where:{id:$%s}){updateProjectResourceTags(input:{%s:$%s}){id}}
-        }""" % (
+        query_str = """mutation AttatchProjectResourceTagsPyApi($%s:ID!,$%s:[String!]) {
+            project(where:{id:$%s}){updateProjectResourceTags(input:{%s:$%s}){id}}}""" % (
             project_id_param,
             tag_ids_param,
             project_id_param,

labelbox/schema/resource_tag.py

@@ -1,7 +1,7 @@
-from labelbox.orm.db_object import DbObject
+from labelbox.orm.db_object import DbObject, Updateable
 from labelbox.orm.model import Field, Relationship
 
-class ResourceTag(DbObject):
+class ResourceTag(DbObject, Updateable):
     """ Resource tag to label and identify your labelbox resources easier.
 
     Attributes:

tests/integration/test_project.py

@@ -42,9 +42,47 @@ def test_project(client, rand_gen):
     assert project not in final
     assert set(final) == set(before)
 
-    # TODO this should raise ResourceNotFoundError, but it doesn't
-    project = client.get_project(project.uid)
+def test_create_project_resource_tags(client, rand_gen):
+    before = list(client.get_projects())
+    for o in before:
+        assert isinstance(o, Project)
+
+    colorA = "#ffffff"
+    textA = rand_gen(str)
+    tag = {"text": textA, "color": colorA}
+
+    colorB = colorA
+    textB = rand_gen(str)
+    tagB = {"text": textB, "color": colorB}
+
+    project_name = rand_gen(str)
+
+    tagA = client.create_resource_tag(tag)
+    assert tagA.text == textA
+    assert '#' + tagA.color == colorA
+    assert tagA.uid is not None
+
+    org = client.get_organization()
+    tags = org.get_resource_tags()
+    lenA = len(tags)
+    assert lenA > 0
+
+    tagB = client.create_resource_tag(tagB)
+    assert tagB.text == textB
+    assert '#' + tagB.color == colorB
+    assert tagB.uid is not None
+
+    tags = org.get_resource_tags()
+    lenB = len(tags)
+    assert lenB > 0
+    assert lenB > lenA
+
+    p1 = client.create_project(name=project_name)
+    assert p1.uid is not None
 
+    project_resource_tag = p1.create_project_resource_tags([tagA.uid])
+    assert len(project_resource_tag) == 1
+    assert project_resource_tag[0].get("id") == tagA.uid
 
 def test_project_filtering(client, rand_gen):
     name_1 = rand_gen(str)