1# Copyright (c) 2015 Amazon.com, Inc. or its affiliates. All Rights Reserved 2# 3# Permission is hereby granted, free of charge, to any person obtaining a 4# copy of this software and associated documentation files (the 5# "Software"), to deal in the Software without restriction, including 6# without limitation the rights to use, copy, modify, merge, publish, dis- 7# tribute, sublicense, and/or sell copies of the Software, and to permit 8# persons to whom the Software is furnished to do so, subject to the fol- 9# lowing conditions: 10# 11# The above copyright notice and this permission notice shall be included 12# in all copies or substantial portions of the Software. 13# 14# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 15# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- 16# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT 17# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, 18# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 19# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS 20# IN THE SOFTWARE. 21# 22 23from boto.compat import json 24from boto.machinelearning.layer1 import MachineLearningConnection 25from tests.unit import AWSMockServiceTestCase 26 27 28class TestMachineLearning(AWSMockServiceTestCase): 29 connection_class = MachineLearningConnection 30 31 def test_predict(self): 32 ml_endpoint = 'mymlmodel.amazonaws.com' 33 self.set_http_response(status_code=200, body=b'') 34 self.service_connection.predict( 35 ml_model_id='foo', record={'Foo': 'bar'}, 36 predict_endpoint=ml_endpoint) 37 self.assertEqual(self.actual_request.host, ml_endpoint) 38 39 def test_predict_with_scheme_in_endpoint(self): 40 ml_endpoint = 'mymlmodel.amazonaws.com' 41 self.set_http_response(status_code=200, body=b'') 42 self.service_connection.predict( 43 ml_model_id='foo', record={'Foo': 'bar'}, 44 predict_endpoint='https://' + ml_endpoint) 45 self.assertEqual(self.actual_request.host, ml_endpoint) 46