list models methods and tests

Author: g-may

test/test_natural_language_understanding.py

@@ -122,3 +122,28 @@ def test_url_analyze(self):
                      features.Emotion(document=False)], url="http://cnn.com",
                     xpath="/bogus/xpath", language="en")
         assert len(responses.calls) == 1
+
+    @responses.activate
+    def test_listModels(self):
+        nlu_url = "http://bogus.com/v1/models"
+        responses.add(responses.GET, nlu_url, status=200, body="{\"resulting_key\": true}",
+                      content_type='application/json')
+        nlu = NaturalLanguageUnderstandingV1(version='2016-01-23',
+                                             url='http://bogus.com',
+                                             username='username',
+                                             password='password')
+        nlu.listModels();
+        assert len(responses.calls) == 1
+
+    @responses.activate
+    def test_deleteModel(self):
+        model_id = "invalid_model_id"
+        nlu_url = "http://bogus.com/v1/models/" + model_id
+        responses.add(responses.DELETE, nlu_url, status=200,
+                      content_type='application/json')
+        nlu = NaturalLanguageUnderstandingV1(version='2016-01-23',
+                                             url='http://bogus.com',
+                                             username='username',
+                                             password='password')
+        nlu.deleteModel(model_id);
+        assert len(responses.calls) == 1

watson_developer_cloud/natural_language_understanding_v1.py

@@ -82,3 +82,24 @@ def analyze(self, features, text=None, url=None, html=None,
                             headers={'content-type': 'application/json'},
                             json=body,
                             accept_json=True)
+    def listModels(self):
+        """
+        Lists the custom models available for your service instance
+        :return: dict of available custom models
+        """
+        return self.request(method='GET', url='/v1/models',
+                            params={"version": self.version},
+                            accept_json=True)
+
+    def deleteModel(self, model_id=None):
+        """
+        Deletes a custom model
+        :param model_id: The ID of the model to delete
+        :return: dict of analyzed text
+        """
+        if model_id is None:
+            raise ValueError("Missing parameter 'model_id'")
+
+        return self.request(method='DELETE', url='/v1/models/'+model_id,
+                            params={"version": self.version},
+                            accept_json=True)