def METHOD_NAME(self, mock_model):
    node = PromptNode()
    assert node.get_prompt_template() is None
    template = node.get_prompt_template("question-answering")
    assert template.name == "question-answering"
    template = node.get_prompt_template(PromptTemplate(name="fake-template", prompt_text=""))
    assert template.name == "fake-template"
    with pytest.raises(ValueError) as e:
        node.get_prompt_template("some-unsupported-template")
        assert e.match("some-unsupported-template not supported, select one of:")
    fake_yaml_prompt = "name: fake-yaml-template\nprompt_text: fake prompt text"
    template = node.get_prompt_template(fake_yaml_prompt)
    assert template.name == "fake-yaml-template"
    fake_yaml_prompt = "- prompt_text: fake prompt text"
    template = node.get_prompt_template(fake_yaml_prompt)
    assert template.name == "custom-at-query-time"
    template = node.get_prompt_template("some prompt")
    assert template.name == "custom-at-query-time"