def test_model_create_transformer_base_name(sagemaker_session):
model = Model(MODEL_IMAGE, MODEL_DATA, sagemaker_session=sagemaker_session)
base_name = "foo"
model._base_name = base_name
transformer = model.transformer(1, "ml.m4.xlarge")
assert base_name == transformer.base_transform_job_name
def test_deploy_generates_endpoint_name_each_time_from_base_name(
production_variant, base_from_name, name_from_base, sagemaker_session):
model = Model(MODEL_IMAGE,
MODEL_DATA,
role=ROLE,
sagemaker_session=sagemaker_session)
base_name = "foo"
model._base_name = base_name
model.deploy(
instance_type=INSTANCE_TYPE,
initial_instance_count=INSTANCE_COUNT,
)
model.deploy(
instance_type=INSTANCE_TYPE,
initial_instance_count=INSTANCE_COUNT,
)
base_from_name.assert_not_called()
name_from_base.assert_called_with(base_name)
assert 2 == name_from_base.call_count
