Python SessionCreateEndpointImageURIRenamer示例

示例#1

文件： test_session_image_uri.py 项目： xiaoyi-cheng/sagemaker-python-sdk

def test_create_endpoint_modify_node():
    modifier = renamed_params.SessionCreateEndpointImageURIRenamer()

    for template in CREATE_ENDPOINT_TEMPLATES:
        call = ast_call(template.format("deployment_image=my_image"))
        modifier.modify_node(call)

        expected = template.format("image_uri=my_image")
        assert expected == pasta.dump(call)

示例#2

文件： test_session_image_uri.py 项目： xiaoyi-cheng/sagemaker-python-sdk

def test_create_endpoint_node_should_be_modified_random_function_call():
    modifier = renamed_params.SessionCreateEndpointImageURIRenamer()
    assert not modifier.node_should_be_modified(ast_call("create_endpoint()"))

示例#3

文件： test_session_image_uri.py 项目： xiaoyi-cheng/sagemaker-python-sdk

def test_create_endpoint_node_should_be_modified_no_image():
    modifier = renamed_params.SessionCreateEndpointImageURIRenamer()

    for template in CREATE_ENDPOINT_TEMPLATES:
        call = ast_call(template.format(""))
        assert not modifier.node_should_be_modified(call)