Index: third_party/gsutil/third_party/boto/tests/unit/machinelearning/test_machinelearning.py |
diff --git a/third_party/gsutil/third_party/boto/tests/unit/machinelearning/test_machinelearning.py b/third_party/gsutil/third_party/boto/tests/unit/machinelearning/test_machinelearning.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..6a4d97c09c2f3a5d6e02eb8b231e9aee330a4e56 |
--- /dev/null |
+++ b/third_party/gsutil/third_party/boto/tests/unit/machinelearning/test_machinelearning.py |
@@ -0,0 +1,45 @@ |
+# Copyright (c) 2015 Amazon.com, Inc. or its affiliates. All Rights Reserved |
+# |
+# Permission is hereby granted, free of charge, to any person obtaining a |
+# copy of this software and associated documentation files (the |
+# "Software"), to deal in the Software without restriction, including |
+# without limitation the rights to use, copy, modify, merge, publish, dis- |
+# tribute, sublicense, and/or sell copies of the Software, and to permit |
+# persons to whom the Software is furnished to do so, subject to the fol- |
+# lowing conditions: |
+# |
+# The above copyright notice and this permission notice shall be included |
+# in all copies or substantial portions of the Software. |
+# |
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- |
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT |
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
+# IN THE SOFTWARE. |
+# |
+ |
+from boto.compat import json |
+from boto.machinelearning.layer1 import MachineLearningConnection |
+from tests.unit import AWSMockServiceTestCase |
+ |
+ |
+class TestMachineLearning(AWSMockServiceTestCase): |
+ connection_class = MachineLearningConnection |
+ |
+ def test_predict(self): |
+ ml_endpoint = 'mymlmodel.amazonaws.com' |
+ self.set_http_response(status_code=200, body=b'') |
+ self.service_connection.predict( |
+ ml_model_id='foo', record={'Foo': 'bar'}, |
+ predict_endpoint=ml_endpoint) |
+ self.assertEqual(self.actual_request.host, ml_endpoint) |
+ |
+ def test_predict_with_scheme_in_endpoint(self): |
+ ml_endpoint = 'mymlmodel.amazonaws.com' |
+ self.set_http_response(status_code=200, body=b'') |
+ self.service_connection.predict( |
+ ml_model_id='foo', record={'Foo': 'bar'}, |
+ predict_endpoint='https://' + ml_endpoint) |
+ self.assertEqual(self.actual_request.host, ml_endpoint) |