def test_delete_model(sagemaker_session): model = Model(MODEL_IMAGE, MODEL_DATA, name=MODEL_NAME, sagemaker_session=sagemaker_session) model.delete_model() sagemaker_session.delete_model.assert_called_with(model.name)
def test_delete_model_no_name(sagemaker_session): model = Model(MODEL_IMAGE, MODEL_DATA, sagemaker_session=sagemaker_session) with pytest.raises( ValueError, match="The SageMaker model must be created first before attempting to delete." ): model.delete_model() sagemaker_session.delete_model.assert_not_called()