fix tests

Author: tomislav-peharda

labelbox/schema/model.py

@@ -42,11 +42,10 @@ def delete_model(self):
             Query execution success.
         """
         ids_param = "ids"
-        query_str = """mutation DeleteModelPyApi($%s: [ID!]! {
-            deleteModels(where: {ids: $%s}) {%s}}""" % (
+        query_str = """mutation DeleteModelPyApi($%s: ID!) {
+            deleteModels(where: {ids: [$%s]})}""" % (
             ids_param, ids_param
         )
-        res = self.client.execute(query_str, {
-            ids_param: self.uid
+        self.client.execute(query_str, {
+            ids_param: str(self.uid)
         })
-        return res

tests/integration/test_model.py

@@ -18,3 +18,14 @@ def test_model(client, configured_project, rand_gen):
 
     model = client.get_model(model.uid)
     assert model.name == data["name"]
+
+
+def test_model_delete(client):
+    before = list(client.get_models())
+
+    model = before[0]
+    model.delete_model()
+
+    after = list(client.get_models())
+
+    assert len(before) == len(after) + 1