def test_image_model_package(self):
arguments = [
'--region', 'us-west-2', '--model_name', 'model_test', '--role',
'arn:aws:iam::123456789012:user/Development/product_1234/*'
]
# does not error out
_utils.create_model_request(
vars(
self.parser.parse_args(arguments + [
'--image', 'test-image', '--model_artifact_url',
's3://fake-bucket/model_artifact'
])))
# does not error out
_utils.create_model_request(
vars(
self.parser.parse_args(arguments +
['--model_package', 'fake-package'])))
with self.assertRaises(Exception):
_utils.create_model_request(vars(
self.parser.parse_args(arguments)))
with self.assertRaises(Exception):
_utils.create_model_request(
vars(
self.parser.parse_args(arguments +
['--image', 'test-image'])))
with self.assertRaises(Exception):
_utils.create_model_request(
vars(
self.parser.parse_args(arguments + [
'--model_artifact_url',
's3://fake-bucket/model_artifact'
])))
def test_pass_most_arguments(self):
arguments = required_args + [
'--container_host_name', 'fake-host', '--tags',
'{"fake_key": "fake_value"}', '--vpc_security_group_ids',
'fake-ids', '--vpc_subnets', 'fake-subnets'
]
response = _utils.create_model_request(
vars(self.parser.parse_args(arguments)))
print(response)
self.assertEqual(
response, {
'ModelName': 'model_test',
'PrimaryContainer': {
'ContainerHostname': 'fake-host',
'Image': 'test-image',
'ModelDataUrl': 's3://fake-bucket/model_artifact',
'Environment': {}
},
'ExecutionRoleArn':
'arn:aws:iam::123456789012:user/Development/product_1234/*',
'Tags': [{
'Key': 'fake_key',
'Value': 'fake_value'
}],
'VpcConfig': {
'SecurityGroupIds': ['fake-ids'],
'Subnets': ['fake-subnets']
},
'EnableNetworkIsolation': True
})
def test_secondary_containers(self):
arguments = required_args + [
'--secondary_containers', '["fake-container"]'
]
response = _utils.create_model_request(
vars(self.parser.parse_args(arguments)))
self.assertEqual(response['Containers'], ['fake-container'])
def test_empty_string(self):
args = self.parser.parse_args(required_args)
response = _utils.create_model_request(vars(args))
test_utils.check_empty_string_values(response)
def test_sample(self):
args = self.parser.parse_args(required_args)
response = _utils.create_model_request(vars(args))
self.assertEqual(response['ModelName'], 'model_test')
