Merge branch 'master' into fix-add-custom-word-method

Author: germanattanasio

test/test_speech_to_text_v1.py

@@ -65,6 +65,8 @@ def _decode_body(body):
 @responses.activate
 def test_custom_model():
     customization_url = 'https://stream.watsonplatform.net/speech-to-text/api/v1/customizations'
+    train_url = "{0}/{1}/train".format(customization_url,'customid')
+
     responses.add(responses.GET, customization_url,
                   body='{"get response": "yep"}', status=200,
                   content_type='application/json')
@@ -81,6 +83,9 @@ def test_custom_model():
                   body='{"bogus_response": "yep"}', status=200,
                   content_type='application/json')
 
+    responses.add(responses.POST, train_url, body='{"bogus_response": "yep"}',
+                  status=200, content_type='application/json')
+
     speech_to_text = watson_developer_cloud.SpeechToTextV1(
         username="username", password="password")
 
@@ -99,10 +104,11 @@ def test_custom_model():
     assert parsed_body['description'] == ''
     assert parsed_body['base_model_name'] == 'en-US_BroadbandModel'
 
+    speech_to_text.train_custom_model('customid')
     speech_to_text.get_custom_model(modelid='modelid')
     speech_to_text.delete_custom_model(modelid='modelid')
 
-    assert len(responses.calls) == 5
+    assert len(responses.calls) == 6
 
 
 def test_custom_corpora():

watson_developer_cloud/speech_to_text_v1.py

@@ -86,6 +86,20 @@ def create_custom_model(self, name, description="",
                             headers={'content-type': 'application/json'},
                             data=json_body, accept_json=True)
 
+    def train_custom_model(self, customization_id,
+                           customization_weight=None,
+                           word_type=None):
+        """
+        Trains a custom language model
+        """
+        params = {'customization_weight': customization_weight,
+                  'word_type': word_type}
+
+        return self.request(method='POST',
+                            url=('/v1/customizations/{0}/train'
+                                 .format(customization_id)), params=params,
+                            accept_json=True)
+
     def list_custom_models(self):
         return self.request(method='GET', url='/v1/customizations',
                             accept_json=True)