ibm_watson 0.1.0

data/test/unit/test_vcap_using_personality_insights.rb

@@ -0,0 +1,150 @@
+# frozen_string_literal: true
+
+require("json")
+require_relative("./../test_helper.rb")
+require("webmock/minitest")
+
+WebMock.disable_net_connect!(allow_localhost: true)
+
+ENV["VCAP_SERVICES"] = JSON.parse(File.read(Dir.getwd + "/resources/vcap-testing.json")).to_json
+
+# Unit tests for VCAP Services using the Personality Insights V3 Service
+class VcapPersonalityInsightsV3Test < Minitest::Test
+  def test_plain_to_json
+    profile_response = File.read(Dir.getwd + "/resources/personality-v3-expect1.txt")
+    personality_text = File.read(Dir.getwd + "/resources/personality-v3.txt")
+    headers = {
+      "Content-Type" => "application/json"
+    }
+    expected_response = DetailedResponse.new(status: 200, body: JSON.parse(profile_response), headers: headers)
+    stub_request(:post, "https://gateway.watsonplatform.net/personality-insights/api/v3/profile?version=2017-10-13")
+      .with(
+        body: personality_text,
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Content-Type" => "text/plain;charset=utf-8",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: profile_response, headers: headers)
+    service = IBMWatson::PersonalityInsightsV3.new(
+      version: "2017-10-13"
+    )
+    service_response = service.profile(
+      content: personality_text,
+      content_type: "text/plain;charset=utf-8"
+    )
+    assert_equal(expected_response.status, service_response.status)
+    assert_equal(expected_response.result, service_response.result)
+    expected_response.headers.each_key do |key|
+      assert(service_response.headers.key?(key))
+      assert(expected_response.headers[key] == service_response.headers[key])
+    end
+  end
+
+  def test_json_to_json
+    profile_response = File.read(Dir.getwd + "/resources/personality-v3-expect2.txt")
+    personality_text = File.read(Dir.getwd + "/resources/personality-v3.json")
+    headers = {
+      "Content-Type" => "applicaiton/json"
+    }
+    expected_response = DetailedResponse.new(status: 200, body: profile_response, headers: headers)
+    stub_request(:post, "https://gateway.watsonplatform.net/personality-insights/api/v3/profile?consumption_preferences=true&raw_scores=true&version=2017-10-13")
+      .with(
+        body: personality_text,
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Content-Type" => "application/json",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: profile_response, headers: headers)
+    service = IBMWatson::PersonalityInsightsV3.new(
+      version: "2017-10-13"
+    )
+    service_response = service.profile(
+      content: personality_text,
+      content_type: "application/json",
+      raw_scores: true,
+      consumption_preferences: true
+    )
+    assert_equal(expected_response.status, service_response.status)
+    assert_equal(expected_response.result, service_response.result)
+    expected_response.headers.each_key do |key|
+      assert(service_response.headers.key?(key))
+      assert(expected_response.headers[key] == service_response.headers[key])
+    end
+  end
+
+  def test_json_to_csv
+    profile_response = File.read(Dir.getwd + "/resources/personality-v3-expect3.txt")
+    personality_text = File.read(Dir.getwd + "/resources/personality-v3.json")
+    headers = {
+      "Content-Type" => "application/json"
+    }
+    expected_response = DetailedResponse.new(status: 200, body: profile_response, headers: headers)
+    stub_request(:post, "https://gateway.watsonplatform.net/personality-insights/api/v3/profile?consumption_preferences=true&csv_headers=true&raw_scores=true&version=2017-10-13")
+      .with(
+        body: personality_text,
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Content-Type" => "application/json",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: profile_response.to_json, headers: headers)
+    service = IBMWatson::PersonalityInsightsV3.new(
+      version: "2017-10-13"
+    )
+    service_response = service.profile(
+      content: personality_text,
+      content_type: "application/json",
+      accept: "text/csv",
+      csv_headers: true,
+      raw_scores: true,
+      consumption_preferences: true
+    )
+    assert_equal(expected_response.status, service_response.status)
+    assert_equal(expected_response.result, service_response.result)
+    expected_response.headers.each_key do |key|
+      assert(service_response.headers.key?(key))
+      assert(expected_response.headers[key] == service_response.headers[key])
+    end
+  end
+
+  def test_plain_to_json_es
+    profile_response = JSON.parse(File.read(Dir.getwd + "/resources/personality-v3-expect4.txt"))
+    personality_text = File.read(Dir.getwd + "/resources/personality-v3-es.txt")
+    headers = {
+      "Content-Type" => "application/json"
+    }
+    expected_response = DetailedResponse.new(status: 200, body: profile_response, headers: headers)
+    stub_request(:post, "https://gateway.watsonplatform.net/personality-insights/api/v3/profile?version=2017-10-13")
+      .with(
+        body: personality_text,
+        headers: {
+          "Accept" => "application/json",
+          "Accept-Language" => "es",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Content-Language" => "es",
+          "Content-Type" => "text/plain;charset=utf-8",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: profile_response.to_json, headers: headers)
+    service = IBMWatson::PersonalityInsightsV3.new(
+      version: "2017-10-13"
+    )
+    service_response = service.profile(
+      content: personality_text,
+      content_type: "text/plain;charset=utf-8",
+      content_language: "es",
+      accept_language: "es"
+    )
+    assert_equal(expected_response.status, service_response.status)
+    assert_equal(expected_response.result, service_response.result)
+    expected_response.headers.each_key do |key|
+      assert(service_response.headers.key?(key))
+      assert(expected_response.headers[key] == service_response.headers[key])
+    end
+  end
+end

data/test/unit/test_visual_recognition_v3.rb

@@ -0,0 +1,345 @@
+# frozen_string_literal: true
+
+require("json")
+require_relative("./../test_helper.rb")
+require("webmock/minitest")
+
+WebMock.disable_net_connect!(allow_localhost: true)
+
+# Unit tests for the Visual Recognition V3 Service
+class VisualRecognitionV3Test < Minitest::Test
+  def test_get_classifier
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19"
+    )
+    service._iam_access_token(iam_access_token: "bogus_access_token")
+    service._iam_access_token(iam_access_token: "bogus_access_token")
+    response = {
+      "classifier_id" => "bogusnumber",
+      "name" => "Dog Breeds",
+      "owner" => "58b61352-678c-44d1-9f40-40edf4ea8d19",
+      "status" => "failed",
+      "created" => "2017-08-25T06:39:01.968Z",
+      "classes" => [
+        {
+          "class" => "goldenretriever"
+        }
+      ]
+    }
+    stub_request(:get, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classifiers/bogusnumber?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_classifier(
+      classifier_id: "bogusnumber"
+    )
+    assert_equal(response, service_response.result)
+  end
+
+  def test_delete_classifier
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    stub_request(:delete, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classifiers/bogusnumber?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: { "response" => 200 }.to_json, headers: { "Content-Type" => "applicaton/json" })
+    service_response = service.delete_classifier(
+      classifier_id: "bogusnumber"
+    )
+    assert_nil(service_response)
+  end
+
+  def test_list_classifiers
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    response = {
+      "classifiers" =>
+        [
+          {
+            "classifier_id" => "InsuranceClaims_1362331461",
+            "name" => "Insurance Claims",
+            "status" => "ready"
+          },
+          {
+            "classifier_id" => "DogBreeds_1539707331",
+            "name" => "Dog Breeds",
+            "status" => "ready"
+          }
+        ]
+    }
+    stub_request(:get, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classifiers?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_classifiers
+    assert_equal(response, service_response.result)
+  end
+
+  def test_create_classifier
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    response = {
+      "classifier_id" => "DogBreeds_2014254824",
+      "name" => "Dog Breeds",
+      "owner" => "58b61352-678c-44d1-9f40-40edf4ea8d19",
+      "status" => "failed",
+      "created" => "2017-08-25T06:39:01.968Z",
+      "classes" => [{ "class" => "goldenretriever" }]
+    }
+    cars = File.open(Dir.getwd + "/resources/cars.zip")
+    trucks = File.open(Dir.getwd + "/resources/trucks.zip")
+
+    stub_request(:post, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classifiers?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.create_classifier(
+      name: "Cars vs Trucks",
+      classname_positive_examples: cars,
+      negative_examples: trucks
+    )
+    assert_equal(response, service_response.result)
+
+    service_response = service.create_classifier(
+      name: "Cars vs Trucks",
+      classname_positive_examples: "cars",
+      classname_positive_examples_filename: "cars",
+      negative_examples: "trucks",
+      negative_examples_filename: "trucks"
+    )
+    assert_equal(response, service_response.result)
+  end
+
+  def test_update_classifier
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    response = {
+      "classifier_id" => "bogusid",
+      "name" => "Insurance Claims",
+      "owner" => "58b61352-678c-44d1-9f40-40edf4ea8d19",
+      "status" => "ready",
+      "created" => "2017-07-17T22:17:14.860Z",
+      "classes" => [
+        { "class" => "motorcycleaccident" },
+        { "class" => "flattire" },
+        { "class" => "brokenwinshield" }
+      ]
+    }
+    stub_request(:post, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classifiers/bogusid?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.update_classifier(
+      classifier_id: "bogusid",
+      classname_positive_examples: "positive examples classname",
+      negative_examples: "negative examples"
+    )
+    assert_equal(response, service_response.result)
+
+    service_response = service.update_classifier(
+      classifier_id: "bogusid",
+      classname_positive_examples: "positive examples file",
+      classname_positive_examples_filename: "positive_filename",
+      negative_examples: "negative examples",
+      negative_examples_filename: "negative_filename"
+    )
+    assert_equal(response, service_response.result)
+  end
+
+  def test_classify
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    response = {
+      "images" =>
+        [
+          {
+            "image" => "test.jpg",
+            "classifiers" =>
+              [
+                {
+                  "classes" =>
+                  [
+                    { "score" => 0.95, "class" => "tiger", "type_hierarchy" => "/animal/mammal/carnivore/feline/big cat/tiger" },
+                    { "score" => 0.997, "class" => "big cat" },
+                    { "score" => 0.998, "class" => "feline" },
+                    { "score" => 0.998, "class" => "carnivore" },
+                    { "score" => 0.998, "class" => "mammal" },
+                    { "score" => 0.999, "class" => "animal" }
+                  ],
+                  "classifier_id" => "default",
+                  "name" => "default"
+                }
+              ]
+          }
+        ],
+      "custom_classes" => 0,
+      "images_processed" => 1
+    }
+
+    stub_request(:post, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classify?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.classify(
+      url: "http://google.com"
+    )
+    assert_equal(response, service_response.result)
+
+    service_response = service.classify(
+      url: "http://google.com",
+      classifier_ids: %w[one two three]
+    )
+    assert_equal(response, service_response.result)
+
+    service_response = service.classify(
+      url: "http://google.com",
+      owners: %w[me IBM]
+    )
+    assert_equal(response, service_response.result)
+
+    image_file = File.open(Dir.getwd + "/resources/test.jpg")
+    service_response = service.classify(
+      images_file: image_file,
+      images_filename: "test.jpg"
+    )
+    assert_equal(response, service_response.result)
+
+    service_response = service.classify(
+      images_file: "image_file"
+    )
+    assert_equal(response, service_response.result)
+  end
+
+  def test_detect_faces
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    response = {
+      "images" => [
+        {
+          "faces" => [
+            {
+              "age" => {
+                "max" => 44,
+                "min" => 35,
+                "score" => 0.446989
+              },
+              "face_location" => {
+                "height" => 159,
+                "left" => 256,
+                "top" => 64,
+                "width" => 92
+              },
+              "gender" => {
+                "gender" => "MALE",
+                "score" => 0.99593
+              },
+              "identity" => {
+                "name" => "Barack Obama",
+                "score" => 0.970688,
+                "type_hierarchy" => "/people/politicians/democrats/barack obama"
+              }
+            }
+          ],
+          "resolved_url" => "https://watson-developer-cloud.github.io/doc-tutorial-downloads/visual-recognition/prez.jpg",
+          "source_url" => "https://watson-developer-cloud.github.io/doc-tutorial-downloads/visual-recognition/prez.jpg"
+        }
+      ],
+      "images_processed" => 1
+    }
+    stub_request(:post, "https://gateway.watsonplatform.net/visual-recognition/api/v3/detect_faces?version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.detect_faces(
+      url: "http://google.com"
+    )
+    assert_equal(response, service_response.result)
+
+    image_file = File.open(Dir.getwd + "/resources/test.jpg")
+    service_response = service.detect_faces(
+      images_file: image_file,
+      images_filename: "test.jpg"
+    )
+    assert_equal(response, service_response.result)
+
+    service_response = service.detect_faces(
+      images_file: "image_file"
+    )
+    assert_equal(response, service_response.result)
+  end
+
+  def test_delete_user_data
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    stub_request(:delete, "https://gateway.watsonplatform.net/visual-recognition/api/v3/user_data?customer_id=id&version=2018-03-19")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "gateway.watsonplatform.net",
+          "Authorization" => "Bearer bogus_access_token"
+        }
+      ).to_return(status: 200, body: "", headers: {})
+    service_response = service.delete_user_data(
+      customer_id: "id"
+    )
+    assert_nil(service_response)
+  end
+
+  def test_get_core_ml_model
+    service = IBMWatson::VisualRecognitionV3.new(
+      version: "2018-03-19",
+      iam_access_token: "bogus_access_token"
+    )
+    stub_request(:get, "https://gateway.watsonplatform.net/visual-recognition/api/v3/classifiers/classifierid/core_ml_model?version=2018-03-19")
+      .with(
+        headers: {
+          "Authorization" => "Bearer bogus_access_token",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: "ml_model", headers: {})
+    service_response = service.get_core_ml_model(classifier_id: "classifierid")
+    assert_equal("ml_model", service_response.result)
+  end
+end