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')