ibm_watson 2.0.1 → 2.1.2

data/lib/ibm_watson/tone_analyzer_v3.rb

@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-# IBM OpenAPI SDK Code Generator Version: 3.19.0-be3b4618-20201113-200858
+# IBM OpenAPI SDK Code Generator Version: 3.31.0-902c9336-20210504-161156
 #
 # The IBM Watson™ Tone Analyzer service uses linguistic analysis to detect
 # emotional and language tones in written text. The service can analyze tone at both the
@@ -34,7 +34,6 @@ require "json"
 require "ibm_cloud_sdk_core"
 require_relative "./common.rb"
 
-# Module for the Watson APIs
 module IBMWatson
   ##
   # The Tone Analyzer V3 service.

data/lib/ibm_watson/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module IBMWatson
-  VERSION = "2.0.1"
+  VERSION = "2.1.2"
 end

data/lib/ibm_watson/visual_recognition_v3.rb

@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-# IBM OpenAPI SDK Code Generator Version: 3.19.0-be3b4618-20201113-200858
+# IBM OpenAPI SDK Code Generator Version: 3.31.0-902c9336-20210504-161156
 #
 # IBM Watson™ Visual Recognition is discontinued. Existing instances are supported
 # until 1 December 2021, but as of 7 January 2021, you can't create instances. Any
@@ -31,7 +31,6 @@ require "json"
 require "ibm_cloud_sdk_core"
 require_relative "./common.rb"
 
-# Module for the Watson APIs
 module IBMWatson
   ##
   # The Visual Recognition V3 service.

data/lib/ibm_watson/visual_recognition_v4.rb

@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-# IBM OpenAPI SDK Code Generator Version: 3.19.0-be3b4618-20201113-200858
+# IBM OpenAPI SDK Code Generator Version: 3.31.0-902c9336-20210504-161156
 #
 # IBM Watson™ Visual Recognition is discontinued. Existing instances are supported
 # until 1 December 2021, but as of 7 January 2021, you can't create instances. Any
@@ -30,7 +30,6 @@ require "json"
 require "ibm_cloud_sdk_core"
 require_relative "./common.rb"
 
-# Module for the Watson APIs
 module IBMWatson
   ##
   # The Visual Recognition V4 service.
@@ -159,7 +158,7 @@ module IBMWatson
     #########################
 
     ##
-    # @!method create_collection(name: nil, description: nil)
+    # @!method create_collection(name: nil, description: nil, training_status: nil)
     # Create a collection.
     # Create a collection that can be used to store images.
     #
@@ -171,8 +170,9 @@ module IBMWatson
     # @param name [String] The name of the collection. The name can contain alphanumeric, underscore, hyphen,
     #   and dot characters. It cannot begin with the reserved prefix `sys-`.
     # @param description [String] The description of the collection.
+    # @param training_status [TrainingStatus] Training status information for the collection.
     # @return [IBMCloudSdkCore::DetailedResponse] A `IBMCloudSdkCore::DetailedResponse` object representing the response.
-    def create_collection(name: nil, description: nil)
+    def create_collection(name: nil, description: nil, training_status: nil)
       raise ArgumentError.new("version must be provided") if version.nil?
 
       headers = {
@@ -186,7 +186,8 @@ module IBMWatson
 
       data = {
         "name" => name,
-        "description" => description
+        "description" => description,
+        "training_status" => training_status
       }
 
       method_url = "/v4/collections"
@@ -264,7 +265,7 @@ module IBMWatson
     end
 
     ##
-    # @!method update_collection(collection_id:, name: nil, description: nil)
+    # @!method update_collection(collection_id:, name: nil, description: nil, training_status: nil)
     # Update a collection.
     # Update the name or description of a collection.
     #
@@ -274,8 +275,9 @@ module IBMWatson
     # @param name [String] The name of the collection. The name can contain alphanumeric, underscore, hyphen,
     #   and dot characters. It cannot begin with the reserved prefix `sys-`.
     # @param description [String] The description of the collection.
+    # @param training_status [TrainingStatus] Training status information for the collection.
     # @return [IBMCloudSdkCore::DetailedResponse] A `IBMCloudSdkCore::DetailedResponse` object representing the response.
-    def update_collection(collection_id:, name: nil, description: nil)
+    def update_collection(collection_id:, name: nil, description: nil, training_status: nil)
       raise ArgumentError.new("version must be provided") if version.nil?
 
       raise ArgumentError.new("collection_id must be provided") if collection_id.nil?
@@ -291,7 +293,8 @@ module IBMWatson
 
       data = {
         "name" => name,
-        "description" => description
+        "description" => description,
+        "training_status" => training_status
       }
 
       method_url = "/v4/collections/%s" % [ERB::Util.url_encode(collection_id)]

data/test/integration/test_discovery_v2.rb

@@ -120,6 +120,7 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
       )
       query_id = service_response.result["query_id"]
       refute(service_response.nil?)
+      # puts JSON.pretty_generate(service_response.result)
 
       service_response = service.get_training_query(
         project_id: @project_id,
@@ -134,6 +135,12 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
         examples: []
       )
       refute(service_response.nil?)
+
+      service_response = service.delete_training_query(
+        project_id: @project_id,
+        query_id: query_id
+      )
+      assert(service_response.nil?)
     end
 
     def test_create_get_update_delete_collection
@@ -238,6 +245,14 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
       assert(service_response.nil?)
     end
 
+    def test_query_collection_notices
+      service_response = service.query_collection_notices(
+        project_id: @project_id,
+        collection_id: @collection_id
+      )
+      refute(service_response.result.nil?)
+    end
+
     def test_delete_user_data
       skip "Covered with the unit test.  No need to run it here"
 

data/test/integration/test_natural_language_understanding_v1.rb

@@ -86,7 +86,7 @@ if !ENV["NATURAL_LANGUAGE_UNDERSTANDING_APIKEY"].nil? && !ENV["NATURAL_LANGUAGE_
     end
 
     def test_list_models
-      skip
+      # skip
       service.add_default_headers(
         headers: {
           "X-Watson-Learning-Opt-Out" => "1",
@@ -96,6 +96,139 @@ if !ENV["NATURAL_LANGUAGE_UNDERSTANDING_APIKEY"].nil? && !ENV["NATURAL_LANGUAGE_
       service_response = service.list_models
       assert((200..299).cover?(service_response.status))
     end
+
+    def test_check_orphands
+      skip "Use to help delete old models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      service_response = service.list_sentiment_models
+      puts JSON.pretty_generate(service_response.result)
+      service_response = service.list_categories_models
+      puts JSON.pretty_generate(service_response.result)
+      service_response = service.list_classifications_models
+      puts JSON.pretty_generate(service_response.result)
+
+      # service.delete_sentiment_model(model_id: "model_id1")
+      # service.delete_categories_model(model_id: "0122b971-94c9-4468-a98f-930f4ce28c32")
+      # service.delete_classifications_model(model_id: "0122b971-94c9-4468-a98f-930f4ce28c32")
+    end
+
+    def test_sentiment_models
+      # skip "test_sentiment_models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      training_data = File.open(Dir.getwd + "/resources/nlu_sentiment_data.csv")
+      service_response = service.create_sentiment_model(
+        language: "en",
+        training_data: training_data
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.list_sentiment_models
+      model_id = service_response.result["models"][0]["model_id"]
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.get_sentiment_model(model_id: model_id)
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.update_sentiment_model(
+        model_id: model_id,
+        language: "en",
+        training_data: training_data
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.delete_sentiment_model(
+        model_id: model_id
+      )
+      assert((200..299).cover?(service_response.status))
+    end
+
+    def test_categories_models
+      # skip "test_categories_models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      training_data = File.open(Dir.getwd + "/resources/nlu_categories_training.json")
+      service_response = service.create_categories_model(
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.list_categories_models
+      model_id = service_response.result["models"][0]["model_id"]
+      assert((200..299).cover?(service_response.status))
+      # puts JSON.pretty_generate(service_response.result)
+      # puts JSON.pretty_generate(model_id)
+
+      service_response = service.get_categories_model(model_id: model_id)
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.update_categories_model(
+        model_id: model_id,
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.delete_categories_model(
+        model_id: model_id
+      )
+      assert((200..299).cover?(service_response.status))
+    end
+
+    def test_classifications_models
+      # skip "test_classifications_models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      training_data = File.open(Dir.getwd + "/resources/nlu_classifications_training.json")
+      service_response = service.create_classifications_model(
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.list_classifications_models
+      model_id = service_response.result["models"][0]["model_id"]
+      assert((200..299).cover?(service_response.status))
+      # puts JSON.pretty_generate(service_response.result)
+      # puts JSON.pretty_generate(model_id)
+
+      service_response = service.get_classifications_model(model_id: model_id)
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.update_classifications_model(
+        model_id: model_id,
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.delete_classifications_model(
+        model_id: model_id
+      )
+      assert((200..299).cover?(service_response.status))
+    end
   end
 else
   class NaturalLanguageUnderstandingV1Test < Minitest::Test

data/test/integration/test_text_to_speech_v1.rb

@@ -83,6 +83,63 @@ if !ENV["TEXT_TO_SPEECH_APIKEY"].nil? && !ENV["TEXT_TO_SPEECH_URL"].nil?
         customization_id: customization_id
       )
     end
+
+    def test_custom_prompts
+      customization_id = @service.create_custom_model(
+        name: "test_integration_customization",
+        description: "RUBY customization for tests"
+      ).result["customization_id"]
+
+      prompt_metadata = { prompt_text: "Thank you and goodbye" }
+      prompt_json = JSON[prompt_metadata]
+      audio_file = File.open(Dir.getwd + "/resources/speech.wav")
+      prompt = @service.add_custom_prompt(
+        customization_id: customization_id,
+        prompt_id: "rubyprompt_id",
+        metadata: prompt_json,
+        file: audio_file
+      ).result["prompt"]
+      assert(prompt == "Thank you and goodbye")
+
+      prompts = @service.list_custom_prompts(customization_id: customization_id).result["prompts"]
+      refute(prompts.nil?)
+
+      prompt = @service.get_custom_prompt(
+        customization_id: customization_id,
+        prompt_id: "rubyprompt_id"
+      ).result["prompt"]
+      assert(prompt == "Thank you and goodbye")
+
+      @service.delete_custom_prompt(
+        customization_id: customization_id,
+        prompt_id: "rubyprompt_id"
+      )
+
+      @service.delete_custom_model(
+        customization_id: customization_id
+      )
+    end
+
+    def test_speaker_models
+      audio_file = File.open(Dir.getwd + "/resources/speech.wav")
+      @service.create_speaker_model(
+        speaker_name: "RubyMike",
+        audio: audio_file
+      ).result["speaker_model"]
+
+      speakers = @service.list_speaker_models.result["speakers"]
+      speaker_id = speakers[0]["speaker_id"]
+      # puts JSON.pretty_generate(speakers)
+
+      @service.get_speaker_model(
+        speaker_id: speaker_id
+      ).result["models_speaker"]
+
+      # speaker_id = d3d03b69-4035-4420-928d-7ac2e0249829
+      @service.delete_speaker_model(
+        speaker_id: speaker_id
+      )
+    end
   end
 else
   class TextToSpeechV1Test < Minitest::Test

data/test/unit/test_discovery_v2.rb

@@ -467,4 +467,33 @@ class DiscoveryV2Test < Minitest::Test
     )
     assert_nil(service_response)
   end
+
+  def test_delete_training_query
+    stub_request(:delete, "https://api.us-south.discovery.watson.cloud.ibm.com/discovery/api/v2/projects/project/training_data/queries/queryid?version=2018-03-05")
+      .with(
+        headers: {
+          "Host" => "api.us-south.discovery.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { "body" => "hello" }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_training_query(
+      project_id: "project",
+      query_id: "queryid"
+    )
+    assert_nil(service_response)
+  end
+
+  def test_query_collection_notices
+    stub_request(:get, "https://api.us-south.discovery.watson.cloud.ibm.com/discovery/api/v2/projects/project/collections/collection/notices?version=2018-03-05")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "api.us-south.discovery.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { "body" => "hello" }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.query_collection_notices(
+      project_id: "project",
+      collection_id: "collection"
+    )
+    assert_equal({ "body" => "hello" }, service_response.result)
+  end
 end

data/test/unit/test_natural_language_understanding_v1.rb

@@ -144,4 +144,235 @@ class NaturalLanguageUnderstandingV1Test < Minitest::Test
     )
     assert_equal({ "deleted" => model_id }, service_response.result)
   end
+
+  def test_sentiment_models
+    authenticator = IBMWatson::Authenticators::BasicAuthenticator.new(
+      username: "username",
+      password: "password"
+    )
+    service = IBMWatson::NaturalLanguageUnderstandingV1.new(
+      version: "2018-03-16",
+      authenticator: authenticator
+    )
+
+    stub_request(:post, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.create_sentiment_model(
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_sentiment_models
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_sentiment_model(model_id: "model_id")
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:put, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.update_sentiment_model(
+      model_id: "model_id",
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:delete, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_sentiment_model(
+      model_id: "model_id"
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+  end
+
+  def test_categories_models
+    authenticator = IBMWatson::Authenticators::BasicAuthenticator.new(
+      username: "username",
+      password: "password"
+    )
+    service = IBMWatson::NaturalLanguageUnderstandingV1.new(
+      version: "2018-03-16",
+      authenticator: authenticator
+    )
+
+    stub_request(:post, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.create_categories_model(
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_categories_models
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_categories_model(model_id: "model_id")
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:put, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.update_categories_model(
+      model_id: "model_id",
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:delete, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_categories_model(
+      model_id: "model_id"
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+  end
+
+  def test_classifications_models
+    authenticator = IBMWatson::Authenticators::BasicAuthenticator.new(
+      username: "username",
+      password: "password"
+    )
+    service = IBMWatson::NaturalLanguageUnderstandingV1.new(
+      version: "2018-03-16",
+      authenticator: authenticator
+    )
+
+    stub_request(:post, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.create_classifications_model(
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_classifications_models
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_classifications_model(model_id: "model_id")
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:put, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.update_classifications_model(
+      model_id: "model_id",
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:delete, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_classifications_model(
+      model_id: "model_id"
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+  end
 end