Add test

Author: tomislav-peharda

labelbox/schema/model_run.py

@@ -89,22 +89,6 @@ def delete_model_run(self):
             ids_param: str(self.uid)
         })
 
-
-class AnnotationGroup(DbObject):
-    label_id = Field.String("label_id")
-    model_run_id = Field.String("model_run_id")
-    data_row = Relationship.ToOne("DataRow", False, cache=True)
-
-    def __init__(self, client, model_id, *args, **kwargs):
-        super().__init__(client, *args, **kwargs)
-        self.model_id = model_id
-
-    @property
-    def url(self):
-        app_url = self.client.app_url
-        endpoint = f"{app_url}/models/{self.model_id}/{self.model_run_id}/AllDatarowsSlice/{self.uid}?view=carousel"
-        return endpoint
-
     def delete_annotation_groups(self, data_row_ids):
         """ Deletes annotation groups by data row ids for a model run.
 
@@ -115,11 +99,28 @@ def delete_annotation_groups(self, data_row_ids):
         """
         model_run_id_param = "modelRunId"
         data_row_ids_param = "dataRowIds"
-        query_str = """mutation DeleteModelRunDataRowsPyApi($%s: ID!, $%s: [ID!]! {
+        query_str = """mutation DeleteModelRunDataRowsPyApi($%s: ID!, $%s: [ID!]!) {
             deleteModelRunDataRows(where: {modelRunId: $%s, dataRowIds: $%s})}""" % (
             model_run_id_param, data_row_ids_param, model_run_id_param, data_row_ids_param
         )
         self.client.execute(query_str, {
-            model_run_id_param: self.model_run_id,
+            model_run_id_param: self.uid,
             data_row_ids_param: data_row_ids
         })
+
+
+class AnnotationGroup(DbObject):
+    label_id = Field.String("label_id")
+    model_run_id = Field.String("model_run_id")
+    data_row = Relationship.ToOne("DataRow", False, cache=True)
+
+    def __init__(self, client, model_id, *args, **kwargs):
+        super().__init__(client, *args, **kwargs)
+        self.model_id = model_id
+
+    @property
+    def url(self):
+        app_url = self.client.app_url
+        endpoint = f"{app_url}/models/{self.model_id}/{self.model_run_id}/AllDatarowsSlice/{self.uid}?view=carousel"
+        return endpoint
+

tests/integration/mal_and_mea/test_model_run.py

@@ -40,5 +40,17 @@ def test_model_run_delete(client, model_run):
 
 
 def test_model_run_delete(client, model_run_annotation_groups):
-    # TODO
-    pass
+    models = list(client.get_models())
+    model = models[0]
+    model_runs = list(model.model_runs())
+    model_run = model_runs[0]
+
+    before = list(model_run.annotation_groups())
+    annotation_group = before[0]
+    
+    data_row_id = annotation_group.data_row().uid
+    model_run.delete_annotation_groups(data_row_ids=[data_row_id])
+
+    after = list(model_run.annotation_groups())
+    
+    assert len(before) == len(after) + 1